CMakeLists.txt 271 字节
Newer Older
1
op_library(tensorrt_engine_op DEPS tensorrt_engine tensorrt_converter infer_io_utils analysis_helper)
N
nhzlx 已提交
2
file(APPEND ${pybind_file} "USE_NO_KERNEL_OP(tensorrt_engine);\n")
W
Wu Yi 已提交
3 4 5
nv_test(test_tensorrt_engine_op SRCS tensorrt_engine_op_test.cc
  DEPS tensorrt_engine_op
  analysis)