| # Copyright (c) OpenMMLab. All rights reserved. | |
| 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 '' | |