Merge pull request #224 from casparvl/reduce_parallelism_for_rocm

boegel · web-flow · commit 38b213a69fb6 · 2026-05-07T19:07:09.000+02:00
Limit parallelism for ROCm-LLVM to max. 12 cores
diff --git a/eb_hooks.py b/eb_hooks.py
@@ -2156,4 +2156,7 @@ def set_maximum(parallel, max_value):
         CPU_TARGET_NEOVERSE_N1: (divide_by_factor, 2),
         CPU_TARGET_NEOVERSE_V1: (divide_by_factor, 2),
     },
+    'ROCm-LLVM': {
+        '*': (set_maximum, 12),
+    },
 }

Original file line number	Diff line number	Diff line change
`@@ -2156,4 +2156,7 @@ def set_maximum(parallel, max_value):`
`2156`	`2156`	`CPU_TARGET_NEOVERSE_N1: (divide_by_factor, 2),`
`2157`	`2157`	`CPU_TARGET_NEOVERSE_V1: (divide_by_factor, 2),`
`2158`	`2158`	`},`
	`2159`	`+ 'ROCm-LLVM': {`
	`2160`	`+ '*': (set_maximum, 12),`
	`2161`	`+ },`
`2159`	`2162`	`}`