pytorch

1
from setuptools import setup
2
from torch.utils.cpp_extension import BuildExtension, CppExtension
3

4
setup(
5
    name="benchmark_cpp_extension",
6
    ext_modules=[CppExtension("benchmark_cpp_extension", ["extension.cpp"])],
7
    cmdclass={"build_ext": BuildExtension},
8
)
9