31 lines
905 B
Python
31 lines
905 B
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=/softs/gcc/11.2.0/bin/gcc"],
|
|
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"]
|