import sys import os # sugon: adapt for DTK # where ROCM bitcode is installed # DEFAULT_ROCM_BC_PATH = '/opt/rocm/opencl/lib/x86_64/bitcode/' DEFAULT_ROCM_BC_PATH = os.environ.get('ROCM_PATH', '/opt/rocm') + '/amdgcn/bitcode/' ROCM_BC_PATH = os.environ.get("NUMBA_ROCM_BC_PATH", DEFAULT_ROCM_BC_PATH)