|
|
|
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 ''
|
|
|
