Update CMakeLists.txt

7 years ago · 01bda73116
parent 2c2a192eb1
commit 01bda73116
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/inference/tensorrt/plugin/CMakeLists.txt
+++ b/paddle/fluid/inference/tensorrt/plugin/CMakeLists.txt
@ -1,3 +1,3 @@
 nv_library(tensorrt_plugin
-           SRCS trt_plugin.cc split_op_plugin.cu prelu_op_plugin.cu
+           SRCS trt_plugin.cc split_op_plugin.cu prelu_op_plugin.cu elementwise_op_plugin.cu
           DEPS enforce tensorrt_engine)