Update ENV for DTK

numba/roc/hlc/config.py

 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 = '/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)

numba/roc/hlc/hlc.py

@@ -110,9 +110,17 @@ class CmdLine(object):
 
     def __init__(self):
         self._binary_path = os.environ.get('HSAILBIN', None)
+
+        # sugon: LLVM tools depend on DTK
+        self._rocm_llvm_binary_path = os.environ.get('ROCM_PATH', '/opt/rocm') + "/llvm/bin"
+        if not os.path.exists(self._rocm_llvm_binary_path):
+            self._rocm_llvm_binary_path = None
+
         def _setup_path(tool):
             if self._binary_path is not None:
                 return os.path.join(self._binary_path, tool)
+            elif self._rocm_llvm_binary_path is not None:
+                return os.path.join(self._rocm_llvm_binary_path, tool)
             else:
                 binpath = os.path.join(sys.prefix, 'bin', tool)
                 return binpath

numba/roc/hsadrv/driver.py

@@ -36,9 +36,8 @@ def _device_type_to_string(device):
     except IndexError:
         return 'Unknown'
 
-
-DEFAULT_HSA_DRIVER = '/opt/rocm/lib/libhsa-runtime64.so'
-
+# sugon: adapt for DTK
+DEFAULT_HSA_DRIVER = os.environ.get('ROCM_PATH', '/opt/rocm') + '/lib/libhsa-runtime64.so'
 
 def _find_driver():
     envpath = os.environ.get('NUMBA_HSA_DRIVER', DEFAULT_HSA_DRIVER)