File size: 406 Bytes
d2410ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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']
|