From d545a4ea25ac8bef4946aee7b45239f6aa531be8 Mon Sep 17 00:00:00 2001 From: Andrei Cravtov Date: Fri, 15 May 2026 13:42:02 +0100 Subject: [PATCH] fff --- src/exo/worker/engines/mlx/utils_mlx.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/exo/worker/engines/mlx/utils_mlx.py b/src/exo/worker/engines/mlx/utils_mlx.py index d4639694f..30cdcaf93 100644 --- a/src/exo/worker/engines/mlx/utils_mlx.py +++ b/src/exo/worker/engines/mlx/utils_mlx.py @@ -815,9 +815,7 @@ def mlx_force_oom2(bytes_alloc: int = 1024**5): # the default is 1 petabyte lol # allocate a lot z = mx.zeros(shape=(size, size), dtype=mx.float32) - t1 = [ - mx.random.uniform(shape=(size, size), dtype=mx.float32) for _ in range(10) - ] + t1 = [mx.random.uniform(shape=(size, size), dtype=mx.float32) for _ in range(2)] print("t1-run") mx.eval(*t1) print("t1-eval")