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

skip onnx node_tests for tensorrt ep (#19880)

Browse source 
fix build break caused by image update. tensorrt isn't expected to pass
all onnx node tests.
This commit is contained in:
George Wu 2024-03-12 23:35:05 -07:00 committed by GitHub
parent 53de2d8cb0
commit 6579f74af0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
onnxruntime/test/providers/cpu/model_tests.cc
View file

@ -620,6 +620,7 @@ static constexpr ORT_STRING_VIEW provider_name_dml = ORT_TSTR("dml");
std::vector<std::filesystem::path> paths;
for (std::pair<ORT_STRING_VIEW, std::vector<ORT_STRING_VIEW>> kvp : provider_names) {
const ORT_STRING_VIEW provider_name = kvp.first;
// Setup ONNX node tests. The test data is preloaded on our CI build machines.
#if !defined(_WIN32)
ORT_STRING_VIEW node_test_root_path = ORT_TSTR("/data/onnx");
@ -627,7 +628,10 @@ static constexpr ORT_STRING_VIEW provider_name_dml = ORT_TSTR("dml");
ORT_STRING_VIEW node_test_root_path = ORT_TSTR("c:\\local\\data\\onnx");
#endif
for (auto p : kvp.second) {
paths.push_back(ConcatPathComponent(node_test_root_path, p));
// tensorrt ep isn't expected to pass all onnx node tests. exclude and run model tests only.
if (provider_name != provider_name_tensorrt) {
paths.push_back(ConcatPathComponent(node_test_root_path, p));
}
}
// Same as the above, except this one is for large models
@ -646,7 +650,6 @@ static constexpr ORT_STRING_VIEW provider_name_dml = ORT_TSTR("dml");
}
#endif
const ORT_STRING_VIEW provider_name = kvp.first;
std::unordered_set<std::basic_string<ORTCHAR_T>> all_disabled_tests(std::begin(immutable_broken_tests),
std::end(immutable_broken_tests));
if (provider_name == provider_name_cuda) {