CMakeLists.txt 434 字节
Newer Older
N
nhzlx 已提交
1 2
nv_library(tensorrt_engine SRCS engine.cc trt_int8_calibrator.cc DEPS ${GLOB_OPERATOR_DEPS} framework_proto device_context boost)
nv_library(tensorrt_op_teller SRCS op_teller.cc DEPS framework_proto boost)
Y
Yan Chunwei 已提交
3
nv_test(test_tensorrt SRCS test_tensorrt.cc DEPS dynload_cuda device_context dynamic_loader)
Y
Yan Chunwei 已提交
4
nv_test(test_tensorrt_engine SRCS test_engine.cc DEPS dynload_cuda tensorrt_engine)
N
nhzlx 已提交
5
add_subdirectory(plugin)
L
Luo Tao 已提交
6
add_subdirectory(convert)