pytorch

1
from setuptools import setup
2

3
from torch.utils.cpp_extension import BuildExtension, CppExtension
4

5

6
setup(
7
    name="no_python_abi_suffix_test",
8
    ext_modules=[
9
        CppExtension("no_python_abi_suffix_test", ["no_python_abi_suffix_test.cpp"])
10
    ],
11
    cmdclass={"build_ext": BuildExtension.with_options(no_python_abi_suffix=True)},
12
)
13