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
       anchor_generator_op_plugin.cu
       yolo_box_op_plugin.cu
       yolo_box_head_op_plugin.cu
       roi_align_op_plugin.cu
       gather_nd_op_plugin.cu
       mish_op_plugin.cu
       pool3d_op_plugin.cu
       deformable_conv_op_plugin.cu
       matmul_op_int8_plugin.cu
       transformer_input_convert_plugin.cu
       remove_padding_plugin.cu
       recover_padding_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)