scalelsd/base/csrc/__init__.py

from torch.utils.cpp_extension import load 
import glob
import os.path as osp

__this__ = osp.dirname(__file__)

#try:
#    _C = load(name='_C',sources=[
#        osp.join(__this__,'binding.cpp'),
#        osp.join(__this__,'linesegment.cu'),
#    ]
#    )
#except:
#    _C = None
_C = None

# _C = load(name='_C', sources=[osp.join(__this__,'binding.cpp'), osp.join(__this__,'linesegment.cu')])
__all__ = ["_C"]

#_C = load(name='base._C', sources=['lltm_cuda.cpp', 'lltm_cuda_kernel.cu'])