Merge pull request #12811 from luotao1/tensorrt_compiler_bug
fix tensorrt compiler bugrevert-12469-sum_op_dim_fix
commit
2ae885e224
@ -1,4 +1,4 @@
|
|||||||
nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto)
|
nv_library(tensorrt_engine SRCS engine.cc DEPS framework_proto device_context)
|
||||||
nv_test(test_tensorrt SRCS test_tensorrt.cc DEPS dynload_cuda device_context dynamic_loader)
|
nv_test(test_tensorrt SRCS test_tensorrt.cc DEPS dynload_cuda device_context dynamic_loader)
|
||||||
nv_test(test_tensorrt_engine SRCS test_engine.cc DEPS dynload_cuda tensorrt_engine)
|
nv_test(test_tensorrt_engine SRCS test_engine.cc DEPS dynload_cuda tensorrt_engine)
|
||||||
add_subdirectory(convert)
|
add_subdirectory(convert)
|
||||||
|
Loading…
Reference in new issue