27 lines
1 KiB
Python
27 lines
1 KiB
Python
import os
|
|
|
|
from torch.utils.cpp_extension import load
|
|
|
|
_src_path = os.path.dirname(os.path.abspath(__file__))
|
|
_backend = load(name='_pvcnn_backend',
|
|
extra_cflags=['-O3', '-std=c++17'],
|
|
extra_cuda_cflags=['--compiler-bindir=/usr/bin/gcc-8'],
|
|
sources=[os.path.join(_src_path,'src', f) for f in [
|
|
'ball_query/ball_query.cpp',
|
|
'ball_query/ball_query.cu',
|
|
'grouping/grouping.cpp',
|
|
'grouping/grouping.cu',
|
|
'interpolate/neighbor_interpolate.cpp',
|
|
'interpolate/neighbor_interpolate.cu',
|
|
'interpolate/trilinear_devox.cpp',
|
|
'interpolate/trilinear_devox.cu',
|
|
'sampling/sampling.cpp',
|
|
'sampling/sampling.cu',
|
|
'voxelization/vox.cpp',
|
|
'voxelization/vox.cu',
|
|
'bindings.cpp',
|
|
]]
|
|
)
|
|
|
|
__all__ = ['_backend']
|