File size: 1,319 Bytes
434b0b0 | 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 30 31 32 33 34 35 36 37 38 | # python3 setup.py install
from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
import os
from distutils.sysconfig import get_config_vars
(opt,) = get_config_vars("OPT")
os.environ["OPT"] = " ".join(
flag for flag in opt.split() if flag != "-Wstrict-prototypes"
)
setup(
name="pointops",
author="Hengshuang Zhao",
ext_modules=[
CUDAExtension(
"pointops_cuda",
[
"src/pointops_api.cpp",
"src/knnquery/knnquery_cuda.cpp",
"src/knnquery/knnquery_cuda_kernel.cu",
"src/sampling/sampling_cuda.cpp",
"src/sampling/sampling_cuda_kernel.cu",
"src/grouping/grouping_cuda.cpp",
"src/grouping/grouping_cuda_kernel.cu",
"src/interpolation/interpolation_cuda.cpp",
"src/interpolation/interpolation_cuda_kernel.cu",
"src/subtraction/subtraction_cuda.cpp",
"src/subtraction/subtraction_cuda_kernel.cu",
"src/aggregation/aggregation_cuda.cpp",
"src/aggregation/aggregation_cuda_kernel.cu",
],
extra_compile_args={"cxx": ["-g"], "nvcc": ["-O2"]},
)
],
cmdclass={"build_ext": BuildExtension},
)
|