You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
15 lines
331 B
15 lines
331 B
11 months ago
|
from setuptools import setup
|
||
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
||
|
|
||
|
setup(
|
||
|
name='feconv_cuda',
|
||
|
ext_modules=[
|
||
|
CUDAExtension('feconv_cuda', [
|
||
|
'feconv_cuda.cpp',
|
||
|
'feconv_cuda_kernel.cu',
|
||
|
]),
|
||
|
],
|
||
|
cmdclass={
|
||
|
'build_ext': BuildExtension
|
||
|
})
|