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