saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-26 19:52:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix for tensorrt_basic_test not being run. (#792)

Browse source
This commit is contained in:
jywu-msft 2019-04-08 13:18:36 -07:00 committed by GitHub
parent 5cf72030b2
commit d91555f99e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/test/providers/trt/trt_basic_test.cc → onnxruntime/test/providers/tensorrt/tensorrt_basic_test.cc
View file

@ -5,7 +5,7 @@
#include "test/providers/provider_test_utils.h"
#include "test/framework/test_utils.h"
#include "gtest/gtest.h"
#include "core/providers/trt/trt_execution_provider.h"
#include "core/providers/tensorrt/tensorrt_execution_provider.h"
using namespace std;
using namespace ONNX_NAMESPACE;

7
tools/ci_build/build.py
View file

@ -484,7 +484,12 @@ def run_onnxruntime_tests(args, source_dir, ctest_path, build_dir, configs, enab
for config in configs:
log.info("Running tests for %s configuration", config)
cwd = get_config_build_dir(build_dir, config)
dll_path = os.path.join(build_dir, config, "external", "tvm", config) if enable_tvm else None
if enable_tvm:
dll_path = os.path.join(build_dir, config, "external", "tvm", config)
elif enable_tensorrt:
dll_path = os.path.join(args.tensorrt_home, 'lib')
else:
dll_path = None
run_subprocess([ctest_path, "--build-config", config, "--verbose"],
cwd=cwd, dll_path=dll_path)