File size: 523 Bytes
cda88e0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
setup(
name='hshadow',
ext_modules=[
CUDAExtension('hshadow', [
'hshadow_cuda.cpp',
'hshadow_cuda_kernel.cu',
])
],
cmdclass={
'build_ext': BuildExtension
}
)
# setup(
# name='plane_visualize',
# ext_modules=[
# CUDAExtension('plane_visualize', [
# 'plane_visualize.cpp',
# 'plane_visualize_cuda.cu',
# ])
# ],
# cmdclass={
# 'build_ext': BuildExtension
# }
# ) |