Python PyTorch CppExtension用法及代码示例

本文简要介绍python语言中 torch.utils.cpp_extension.CppExtension 的用法。

用法:

torch.utils.cpp_extension.CppExtension(name, sources, *args, **kwargs)

为 C++ 创建一个 setuptools.Extension。

创建setuptools.Extension 的便捷方法，使用最少(但通常足够)的参数来构建 C++ 扩展。

所有参数都转发到setuptools.Extension 构造函数。

示例

>>> from setuptools import setup
>>> from torch.utils.cpp_extension import BuildExtension, CppExtension
>>> setup(
        name='extension',
        ext_modules=[
            CppExtension(
                name='extension',
                sources=['extension.cpp'],
                extra_compile_args=['-g']),
        ],
        cmdclass={
            'build_ext': BuildExtension
        })