10 lines
287 B
Python
10 lines
287 B
Python
from setuptools import setup, Extension
|
|
from torch.utils import cpp_extension
|
|
|
|
setup(
|
|
name='quant_cuda',
|
|
ext_modules=[cpp_extension.CUDAExtension(
|
|
'quant_cuda', ['quant_cuda.cpp', 'quant_cuda_kernel.cu']
|
|
)],
|
|
cmdclass={'build_ext': cpp_extension.BuildExtension}
|
|
)
|