You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
3.4 KiB
53 lines
3.4 KiB
# Add TRT tests
|
|
nv_library(tensorrt_converter
|
|
SRCS mul_op.cc conv2d_op.cc fc_op.cc pool2d_op.cc elementwise_op.cc
|
|
batch_norm_op.cc activation_op.cc softmax_op.cc concat_op.cc dropout_op.cc
|
|
pad_op.cc split_op.cc prelu_op.cc leaky_relu_op.cc gelu_op.cc layer_norm_op.cc multihead_matmul_op.cc
|
|
shuffle_channel_op.cc swish_op.cc instance_norm_op.cc stack_op.cc
|
|
emb_eltwise_layernorm.cc skip_layernorm.cc scale_op.cc slice_op.cc hard_sigmoid_op.cc hard_swish_op.cc
|
|
DEPS tensorrt_engine tensorrt_plugin operator scope framework_proto op_registry)
|
|
|
|
nv_test(test_op_converter SRCS test_op_converter.cc DEPS
|
|
paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_converter)
|
|
|
|
# TODO(xingzhaolong): fix the the following ci ut error.
|
|
|
|
#nv_test(test_io_converter SRCS test_io_converter.cc io_converter.cc DEPS dynload_cuda dynamic_loader lod_tensor)
|
|
#nv_test(test_trt_mul_op SRCS test_mul_op.cc mul_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine mul_op)
|
|
#nv_test(test_trt_fc_op SRCS test_fc_op.cc fc_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine mul_op)
|
|
#nv_test(test_trt_activation_op SRCS test_activation_op.cc activation_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine activation_op)
|
|
#nv_test(test_trt_conv_op SRCS test_conv2d_op.cc conv2d_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine conv_op conv_transpose_op)
|
|
#nv_test(test_trt_pool2d_op SRCS test_pool2d_op.cc pool2d_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine pool_op tensorrt_plugin)
|
|
#nv_test(test_trt_elementwise_op SRCS test_elementwise_op.cc elementwise_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_plugin
|
|
# elementwise_add_op elementwise_mul_op)
|
|
#nv_test(test_trt_softmax_op SRCS test_softmax_op.cc softmax_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine softmax_op)
|
|
#nv_test(test_trt_batch_norm_op SRCS test_batch_norm_op.cc batch_norm_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine batch_norm_op)
|
|
#nv_test(test_trt_concat_op SRCS test_concat_op.cc concat_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine concat_op)
|
|
#nv_test(test_trt_dropout_op SRCS test_dropout_op.cc dropout_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine dropout_op)
|
|
#nv_test(test_trt_pad_op SRCS test_pad_op.cc pad_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine pad_op)
|
|
#nv_test(test_trt_split_op SRCS test_split_op.cc split_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_plugin
|
|
# split_op concat_op)
|
|
#nv_test(test_trt_prelu_op SRCS test_prelu_op.cc prelu_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_plugin
|
|
# prelu_op)
|
|
#nv_test(test_trt_leaky_relu_op SRCS test_leaky_relu_op.cc leaky_relu_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine activation_op)
|
|
|
|
#nv_test(test_shuffle_channel_op SRCS test_shuffle_channel_op.cc shuffle_channel_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine shuffle_channel_op)
|
|
|
|
#nv_test(test_swish_op SRCS test_swish_op.cc swish_op.cc
|
|
# DEPS paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine activation_op tensorrt_plugin)
|