From 787cd28b56351e2cd65cf2800183e1031ac3fdfd Mon Sep 17 00:00:00 2001 From: qingqing01 Date: Tue, 22 Oct 2019 12:12:49 +0800 Subject: [PATCH] Update custom_op.md (#1528) --- doc/fluid/advanced_usage/development/new_op/custom_op.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/doc/fluid/advanced_usage/development/new_op/custom_op.md b/doc/fluid/advanced_usage/development/new_op/custom_op.md index ce8538e21..706399bb0 100644 --- a/doc/fluid/advanced_usage/development/new_op/custom_op.md +++ b/doc/fluid/advanced_usage/development/new_op/custom_op.md @@ -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 -- GitLab