Merge pull request #25 from laraPPr/update_hooks

boegel · web-flow · commit 89b7cbaee5e1 · 2026-06-09T19:43:36.000+02:00
update eb_hooks.py for memory expensive builds of jax
diff --git a/eb_hooks.py b/eb_hooks.py
@@ -2195,4 +2195,7 @@ def set_maximum(parallel, max_value):
     'ROCm-LLVM': {
         '*': (set_maximum, 12),
     },
+    'jax': {
+        '*': (divide_by_factor, 2),
+    },
 }

Original file line number	Diff line number	Diff line change
`@@ -2195,4 +2195,7 @@ def set_maximum(parallel, max_value):`
`2195`	`2195`	`'ROCm-LLVM': {`
`2196`	`2196`	`'*': (set_maximum, 12),`
`2197`	`2197`	`},`
	`2198`	`+ 'jax': {`
	`2199`	`+ '*': (divide_by_factor, 2),`
	`2200`	`+ },`
`2198`	`2201`	`}`