| |
| 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}, |
| ) |
|
|
