CMakeLists.txt 671 字节
Newer Older
Y
Yan Chunwei 已提交
1 2
# Add TRT tests
nv_test(test_op_converter SRCS test_op_converter.cc mul_op.cc conv2d_op.cc DEPS ${FLUID_CORE_MODULES} tensorrt_engine)
Y
Yu Yang 已提交
3 4 5 6 7
# This test is not stable
# See https://paddleci.ngrok.io/viewLog.html?tab=buildLog&buildTypeId=Paddle_PrCi2&buildId=36834&_focus=8828 
#nv_test(test_trt_activation_op SRCS test_activation_op.cc activation_op.cc io_converter.cc
#    DEPS ${FLUID_CORE_MODULES} activation_op tensorrt_engine
#    SERIAL)
L
Luo Tao 已提交
8
nv_test(test_io_converter SRCS test_io_converter.cc io_converter.cc DEPS dynload_cuda dynamic_loader lod_tensor)
Y
Yan Chunwei 已提交
9 10
nv_test(test_trt_mul_op SRCS test_mul_op.cc mul_op.cc
        DEPS ${FLUID_CORE_MODULES} tensorrt_engine mul_op SERIAL)