from setuptools import setup
from torch.utils.cpp_extension import CppExtension, BuildExtension

setup(name='benchmark_cpp_extension',
      ext_modules=[CppExtension('benchmark_cpp_extension', ['extension.cpp'])],
      cmdclass={'build_ext': BuildExtension})