diff --git a/eb_hooks.py b/eb_hooks.py index 33e39388..5388fbbe 100644 --- a/eb_hooks.py +++ b/eb_hooks.py @@ -2904,4 +2904,7 @@ def set_maximum(parallel, max_value): 'jax': { '*': (divide_by_factor, 2), }, + 'polars': { + CPU_TARGET_A64FX: (set_maximum, 4), + }, }