wehub-resource-sync
22 lines
715 B
CMake
22 lines
715 B
CMake
set(TENSORRT_VERSION_NUMBER
|
|
"${TENSORRT_MAJOR_VERSION}${TENSORRT_MINOR_VERSION}")
|
|
if(${TENSORRT_VERSION_NUMBER} GREATER_EQUAL 85)
|
|
nv_test(
|
|
test_tensorrt_engine_instruction
|
|
SRCS test_tensorrt_engine_instruction.cc
|
|
DEPS pir
|
|
trt_engine
|
|
naive_executor
|
|
phi
|
|
common
|
|
pir_save_load
|
|
pir_transforms
|
|
pir_tensorrt_plugin)
|
|
set_tests_properties(test_tensorrt_engine_instruction PROPERTIES TIMEOUT 120)
|
|
if(WITH_ONNXRUNTIME AND WIN32)
|
|
# Copy onnxruntime for some c++ test in Windows, since the test will
|
|
# be build only in CI, so suppose the generator in Windows is Ninja.
|
|
copy_onnx(test_tensorrt_engine_instruction)
|
|
endif()
|
|
endif()
|