CMakeLists.txt

nv_library(tensorrt_plugin
           SRCS trt_plugin.cc split_op_plugin.cu elementwise_op_plugin.cu
           prelu_op_plugin.cu gelu_op_plugin.cu
           pool_op_plugin.cu swish_op_plugin.cu layer_norm_op_plugin.cu
           instance_norm_op_plugin.cu emb_eltwise_layernorm_plugin.cu
           qkv_to_context_plugin.cu skip_layernorm_op_plugin.cu slice_op_plugin.cu
           hard_swish_op_plugin.cu stack_op_plugin.cu special_slice_plugin.cu
           anchor_generator_op_plugin.cu
           yolo_box_op_plugin.cu
           roi_align_op_plugin.cu
           gather_nd_op_plugin.cu
           pool3d_op_plugin.cu
           mish_op_plugin.cu
           DEPS enforce tensorrt_engine prelu tensor bert_encoder_functor)

nv_test(test_split_plugin SRCS test_split_plugin.cc DEPS
  paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_plugin)