From 475e123a817e3c9ee2adb7bbe36e97593b309525 Mon Sep 17 00:00:00 2001
From: Bert Maher <bertrand@fb.com>
Date: Tue, 9 Jul 2024 14:18:51 -0700
Subject: [PATCH] tweak shapes (#2372)

Summary: Pull Request resolved: https://github.com/pytorch/benchmark/pull/2372

Reviewed By: manman-ren, jianyuh

Differential Revision: D59535933

Pulled By: bertmaher

fbshipit-source-id: b361e66e22f2824c542446b83a75061e02e7cbcc
---
 torchbenchmark/util/triton_op.py | 5 -----
 1 file changed, 5 deletions(-)

diff --git a/torchbenchmark/util/triton_op.py b/torchbenchmark/util/triton_op.py
index b180c51c7..fa88e03b8 100644
--- 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)