mirror of
https://github.com/botastic/SoftGroup.git
synced 2025-10-16 11:45:42 +00:00
16 lines
429 B
Python
16 lines
429 B
Python
from setuptools import setup
|
|
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
|
|
|
setup(
|
|
name='SOFTGROUP_OP',
|
|
ext_modules=[
|
|
CUDAExtension(
|
|
'SOFTGROUP_OP', ['src/softgroup_api.cpp', 'src/softgroup_ops.cpp', 'src/cuda.cu'],
|
|
extra_compile_args={
|
|
'cxx': ['-g'],
|
|
'nvcc': ['-O2']
|
|
})
|
|
],
|
|
cmdclass={'build_ext': BuildExtension})
|