| 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=/usr/bin/gcc-8'], | |
| sources=[os.path.join(_src_path, f) for f in ['connected_components.cu']] | |
| ) | |
| __all__ = ['_backend'] | |