belle

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

4
setup(
5
    name='quant_cuda',
6
    ext_modules=[cpp_extension.CUDAExtension(
7
        'quant_cuda', ['quant_cuda.cpp', 'quant_cuda_kernel.cu']
8
    )],
9
    cmdclass={'build_ext': cpp_extension.BuildExtension}
10
)
11