update roc 2

numba/core/config.py

@@ -501,6 +501,19 @@ class _EnvReloader(object):
 
         NUMBA_DEFAULT_NUM_THREADS = num_threads_default()
 
+        # HSA Configs
+
+        # Disable HSA support
+        DISABLE_HSA = _readenv("NUMBA_DISABLE_HSA", int, 0)
+
+        # The default number of threads to use.
+        NUMBA_DEFAULT_NUM_THREADS = max(
+            1,
+            len(os.sched_getaffinity(0))
+            if hasattr(os, "sched_getaffinity")
+            else multiprocessing.cpu_count(),
+        )
+
         # Numba thread pool size (defaults to number of CPUs on the system).
         _NUMBA_NUM_THREADS = _readenv("NUMBA_NUM_THREADS", int,
                                       NUMBA_DEFAULT_NUM_THREADS)