From e9575c81e27403e2435e43526ed4d2830815e542 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Wed, 27 Aug 2025 12:49:53 -0700 Subject: [PATCH] delete --- test/test_rangeify.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_rangeify.py b/test/test_rangeify.py index 0febe622d8..b83dff6261 100644 --- a/test/test_rangeify.py +++ b/test/test_rangeify.py @@ -116,10 +116,10 @@ class TestRangeify(unittest.TestCase): out.realize() def test_flash_attention(self): - BS, HEADS, SEQLEN, EMB = 4, 2, 16, 8 + #BS, HEADS, SEQLEN, EMB = 4, 2, 16, 8 # bigger - #BS, HEADS, SEQLEN, EMB = 4, 32, 128, 64 + BS, HEADS, SEQLEN, EMB = 4, 32, 1024, 64 # llama 8B #BS, HEADS, SEQLEN, EMB = 4, 32, 2048, 128