1from setuptools import setup 2 3from torch.utils.cpp_extension import BuildExtension, CppExtension 4 5 6setup( 7 name="benchmark_cpp_extension", 8 ext_modules=[CppExtension("benchmark_cpp_extension", ["extension.cpp"])], 9 cmdclass={"build_ext": BuildExtension}, 10) 11