config.py 302 Bytes
Newer Older
dugupeiwen's avatar
dugupeiwen committed
1
2
import sys
import os
dugupeiwen's avatar
dugupeiwen committed
3
# sugon: adapt for DTK
dugupeiwen's avatar
dugupeiwen committed
4
5

# where ROCM bitcode is installed
dugupeiwen's avatar
dugupeiwen committed
6
7
# DEFAULT_ROCM_BC_PATH = '/opt/rocm/opencl/lib/x86_64/bitcode/'
DEFAULT_ROCM_BC_PATH = os.environ.get('ROCM_PATH', '/opt/rocm') + '/amdgcn/bitcode/'
dugupeiwen's avatar
dugupeiwen committed
8
ROCM_BC_PATH = os.environ.get("NUMBA_ROCM_BC_PATH", DEFAULT_ROCM_BC_PATH)