PVD/modules/functional/backend.py

31 lines
905 B
Python
Raw Normal View History

2021-10-19 20:54:46 +00:00
import os
from torch.utils.cpp_extension import load
_src_path = os.path.dirname(os.path.abspath(__file__))
2023-04-11 09:12:58 +00:00
_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",
]
],
)
2021-10-19 20:54:46 +00:00
2023-04-11 09:12:58 +00:00
__all__ = ["_backend"]