mirror of
https://github.com/microsoft/BitNet.git
synced 2026-05-03 11:20:36 +00:00
Junhui He
13 lines
299 B
Python
13 lines
299 B
Python
from setuptools import setup
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
|
|
setup(
|
|
name='bitlinear_cpp',
|
|
ext_modules=[
|
|
CUDAExtension('bitlinear_cuda', [
|
|
'bitnet_kernels.cu',
|
|
])
|
|
],
|
|
cmdclass={
|
|
'build_ext': BuildExtension
|
|
}) |