Skip to content

Latest commit

 

History

History
98 lines (67 loc) · 3.1 KB

README.md

File metadata and controls

98 lines (67 loc) · 3.1 KB

自定义OP的编译过程

注意： 请使用4.8及以上版本的gcc进行自定义OP的编译，否则可能引入兼容性问题。

代码结构

  • src: 扩展OP C++/CUDA 源码
  • pointnet_lib.py: Python封装
  • tests: 各OP单测程序

安装PaddlePaddle

请通过如下方式安装PaddlePaddle：

  • 通过Paddle develop分支源码编译安装，编译方法如下:

    1. Ubuntu
    2. CentOS
    3. MasOS
    4. Windows

    说明： 推荐使用docker编译

  • 安装Paddle develop每日版本whl包

    注意： 编译自定义OP使用的gcc版本须与Paddle编译使用gcc版本一致，Paddle develop每日版本目前采用gcc 4.8.2版本编译，若使用每日版本，请使用gcc 4.8.2版本编译自定义OP，否则可能出现兼容性问题。

编译自定义OP

自定义op需要将实现的C++、CUDA代码编译成动态库，mask.sh中通过g++/nvcc编译，当然您也可以写Makefile或者CMake。

编译需要include PaddlePaddle的相关头文件，链接PaddlePaddle的lib库。 头文件和lib库可通过下面命令获取到:

# python
>>> import paddle
>>> print(paddle.sysconfig.get_include())
/paddle/pyenv/local/lib/python2.7/site-packages/paddle/include
>>> print(paddle.sysconfig.get_lib())
/paddle/pyenv/local/lib/python2.7/site-packages/paddle/libs

我们提供动态库编译脚本如下：

cd src
sh make.sh

最终编译会产出pointnet_lib.so

说明： 若使用源码编译安装PaddlePaddle的方式，编译过程中cmake未设置WITH_MKLDNN的方式， 编译自定义OP时会报错找不到mkldnn.h等文件，可在make.sh中删除编译命令中的-DPADDLE_WITH_MKLDNN选项。

设置环境变量

需要将Paddle的核心库设置到LD_LIBRARY_PATH里, 先运行下面程序获取路径:

import paddle
print(paddle.sysconfig.get_lib())

可通过如下方式添加动态库路径:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python -c 'import paddle; print(paddle.sysconfig.get_lib())'`

执行单测

执行下列单测，确保自定义算子可在网络中正确使用：

# 回到 ext_op 目录，添加 PYTHONPATH
cd ..
export PYTHONPATH=$PYTHONPATH:`pwd`

# 运行单测 
python test/test_farthest_point_sampling_op.py
python test/test_gather_point_op.py
python test/test_group_points_op.py
python test/test_query_ball_op.py
python test/test_three_interp_op.py
python test/test_three_nn_op.py

单测运行成功会输出提示信息，如下所示：

.
----------------------------------------------------------------------
Ran 1 test in 13.205s

OK

更多关于如何在框架外部自定义 C++ OP，可阅读官网说明文档