saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add Missing Test File (#13607)

Browse source 
I built a new test infra for CUDA EP in #13016 but forgot adding the
test to onnxruntime_test_all. Here is the missing file. Now, the
`TestAll` function is really called in CI.
This commit is contained in:
Wei-Sheng Chin 2022-11-10 09:56:19 -08:00 committed by GitHub
parent 31cb3cb254
commit cd85a6333a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 25 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/cuda/cuda_provider_factory.cc
View file

@ -200,7 +200,7 @@ struct ProviderInfo_CUDA_Impl : ProviderInfo_CUDA {
// TODO(wechi): brings disabled tests in onnxruntime/test/providers/cuda/*
// back alive here.
return false;
return true;
}
#endif
} g_info;

24
onnxruntime/test/providers/cuda/cuda_provider_test.cc Normal file
View file

@ -0,0 +1,24 @@
// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the MIT License.
#ifndef NDEBUG
#include "gtest/gtest.h"
#include "core/common/status.h"
#include "core/providers/cuda/cuda_provider_factory.h"
#include <iostream>
namespace onnxruntime {
ProviderInfo_CUDA& GetProviderInfo_CUDA();
namespace test {
namespace cuda {
TEST(CUDAEPTEST, ALL) {
onnxruntime::ProviderInfo_CUDA& ep = onnxruntime::GetProviderInfo_CUDA();
ASSERT_TRUE(ep.TestAll());
}
} // namespace test
} // namespace cuda
} // namespace onnxruntime
#endif