|
|
|
import glob |
|
import os |
|
|
|
import torch |
|
|
|
if torch.__version__ == 'parrots': |
|
import parrots |
|
|
|
def get_compiler_version(): |
|
return 'GCC ' + parrots.version.compiler |
|
|
|
def get_compiling_cuda_version(): |
|
return parrots.version.cuda |
|
else: |
|
from ..utils import ext_loader |
|
ext_module = ext_loader.load_ext( |
|
'_ext', ['get_compiler_version', 'get_compiling_cuda_version']) |
|
|
|
def get_compiler_version(): |
|
return ext_module.get_compiler_version() |
|
|
|
def get_compiling_cuda_version(): |
|
return ext_module.get_compiling_cuda_version() |
|
|
|
|
|
def get_onnxruntime_op_path(): |
|
wildcard = os.path.join( |
|
os.path.abspath(os.path.dirname(os.path.dirname(__file__))), |
|
'_ext_ort.*.so') |
|
|
|
paths = glob.glob(wildcard) |
|
if len(paths) > 0: |
|
return paths[0] |
|
else: |
|
return '' |
|
|
