CppExtension

paddle.utils.cpp_extension. CppExtension ( sources, args, kwargs* ) [源代码]

此接口用于配置自定义 OP 的源文件信息，编译生成仅支持 CPU 设备上执行的算子。若要编译同时支持 GPU 设备的算子，请使用 CUDAExtension 。

此接口是对 Python 内建库 setuptools.Extension 的进一步封装。除了不需要显式地指定 name 参数，其他参数以及使用方式上，与原生内建库接口保持一致。

使用样例如下：

  1. # setup.py
  3. # 编译仅支持 CPU 的算子
  4. from paddle.utils.cpp_extension import CppExtension, setup
  6. setup(
  7.     name='custom_op',
  8.     ext_modules=CppExtension(sources=['relu_op.cc'])
  9. )

注解

搭配 setup 接口使用，编译生成的动态库名称与 setup 接口中的 name 一致。

参数：

  • sources (list[str]): 用于指定自定义 OP 对应的源码文件。cpp 源文件支持 .cc、.cpp等后缀

  • *args, kwargs** (可选): 用于指定 Extension 的其他参数，支持的参数与 setuptools.Extension 一致。

返回： setuptools.Extension 对象

使用本API的教程文档