未验证 提交 e8d84133 编写于 作者: Q qingqing01 提交者: GitHub

Update custom_op.md (#1529)

上级 66e90d64
......@@ -258,6 +258,7 @@ nvcc relu_op.cu -c -o relu_op.cu.o -ccbin cc -DPADDLE_WITH_CUDA -DEIGEN_USE_GPU
-I ${include_dir}/third_party/install/xxhash/include \
-I ${include_dir}/third_party/boost \
-I ${include_dir}/third_party/eigen3 \
-I ${include_dir}/third_party/dlpack/include \
-I ${include_dir}/third_party/threadpool/src/extern_threadpool \
-I ${include_dir} \
......@@ -269,6 +270,7 @@ g++ relu_op.cc relu_op.cu.o -o relu2_op.so -shared -fPIC -std=c++11 -O3 \
-I ${include_dir}/third_party/install/zlib/include \
-I ${include_dir}/third_party/boost \
-I ${include_dir}/third_party/eigen3 \
-I ${include_dir}/third_party/dlpack/include \
-I ${include_dir} \
-L /usr/local/cuda/lib64 \
-L ${lib_dir} -lpaddle_framework -lcudart
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册