tweak shapes (#2372)

Summary: Pull Request resolved: #2372 Reviewed By: manman-ren, jianyuh Differential Revision: D59535933 Pulled By: bertmaher fbshipit-source-id: b361e66e22f2824c542446b83a75061e02e7cbcc
pytorch · Jul 9, 2024 · 475e123 · 475e123
1 parent 9783803
commit 475e123
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/torchbenchmark/util/triton_op.py b/torchbenchmark/util/triton_op.py
@@ -132,11 +132,6 @@ def llama_shapes():
         (1024, 8192),
         (8192, 7168),
         (3584, 8192),
-
-        (16384, 2304),
-        (2048, 16384),
-        (16384, 13312),
-        (6656, 16384),
     ]
     return [
         (bs, n, k, None)