请问天元框架中有实现CPP_extension的模块吗

#1

您好，我有一个用C++写的模块，现在想把它以setup.py的方式进行安装，请问天元框架有实现Pytroch中TORCH.UTILS.CPP_EXTENSION 这样的模块吗？
如果我想要实现类似于pytorch 中 CppExtension、CUDAExtension的功能，在天元框架中需要怎么做呢？

>>> 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
        })
#2

不是太明白您的意思？

  • 目前megengine 的python包是prebuild的， 在imperative/python/setup.py
  • 你如果修改或者加了新的CPP， 可以通过检查CMakeLists.txt 是否已经自动包含了你的修改文件，如果已经自动包含了， 你只需要参考scripts/whl/BUILD_PYTHON_WHL_README.md 来更新你的修改就可以了， 否则需要先修改CMakeLists.txt，
  • 因为构建时间太长了， 我们不支持pip install的时候才去构建src
#3

可能是希望要 Custom OP？https://www.megengine.org.cn/doc/stable/zh/user-guide/tools/customop.html