File size: 491 Bytes
4c954ae
 
 
 
 
 
821a6d0
 
 
 
 
 
 
 
 
4c954ae
821a6d0
4c954ae
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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'])