CMakeLists.txt 526 字节
Newer Older
1 2 3 4
set(tensorrt_engine_op_deps tensorrt_engine tensorrt_converter infer_io_utils
                            analysis_helper)

op_library(tensorrt_engine_op DEPS ${tensorrt_engine_op_deps})
5 6 7
nv_test(
  test_tensorrt_engine_op
  SRCS tensorrt_engine_op_test.cc
8
  DEPS tensorrt_engine_op analysis fleet_executor)
9 10 11 12 13 14

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_op)
endif()