34 lines
1006 B
Python
34 lines
1006 B
Python
|
# Copyright (c) Facebook, Inc. and its affiliates.
|
||
|
|
#
|
||
|
|
# This source code is licensed under the MIT license found in the
|
||
|
|
# LICENSE file in the root directory of this source tree.
|
||
|
|
|
||
|
|
from setuptools import setup
|
||
|
|
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
|
||
|
|
import glob
|
||
|
|
import os
|
||
|
|
ROOT = os.path.dirname(os.path.abspath(__file__))
|
||
|
|
|
||
|
|
_ext_src_root = "_ext_src"
|
||
|
|
_ext_sources = glob.glob("{}/src/*.cpp".format(_ext_src_root)) + glob.glob(
|
||
|
|
"{}/src/*.cu".format(_ext_src_root)
|
||
|
|
)
|
||
|
|
_ext_headers = glob.glob("{}/include/*".format(_ext_src_root))
|
||
|
|
|
||
|
|
setup(
|
||
|
|
name='pointnet2',
|
||
|
|
ext_modules=[
|
||
|
|
CUDAExtension(
|
||
|
|
name='pointnet2._ext',
|
||
|
|
sources=_ext_sources,
|
||
|
|
extra_compile_args={
|
||
|
|
"cxx": ["-O2", "-I{}".format("{}/{}/include".format(ROOT, _ext_src_root))],
|
||
|
|
"nvcc": ["-O2", "-I{}".format("{}/{}/include".format(ROOT, _ext_src_root))],
|
||
|
|
},
|
||
|
|
)
|
||
|
|
],
|
||
|
|
cmdclass={
|
||
|
|
'build_ext': BuildExtension
|
||
|
|
}
|
||
|
|
)
|