CMakeLists.txt 433 字节
Newer Older
1
nv_library(tensorrt_plugin
N
nhzlx 已提交
2
           SRCS trt_plugin.cc split_op_plugin.cu elementwise_op_plugin.cu
3 4 5
           prelu_op_plugin.cu  trt_plugin_factory.cc 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
6
qkv_to_context_plugin.cu skip_layernorm_op_plugin.cu
7
           DEPS enforce tensorrt_engine prelu tensor bert_encoder_functor)