From 795255d398a4eb946a9bc074f2e644de036bd5d8 Mon Sep 17 00:00:00 2001 From: Ryan Spring Date: Fri, 30 Aug 2024 17:13:16 -0700 Subject: [PATCH] change matmul size --- tests/python/opinfo_input_generators.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/python/opinfo_input_generators.py b/tests/python/opinfo_input_generators.py index 779f84bfb3b..f7b82d745bb 100644 --- a/tests/python/opinfo_input_generators.py +++ b/tests/python/opinfo_input_generators.py @@ -1492,9 +1492,9 @@ def matmul_input_generator( requires_grad=requires_grad, ) - B = 64 - M = 512 - N = 256 + B = 8 + M = 256 + N = 128 K = 32 shapes_a = ((K,), (M, K), (1, K), (B, M, K), (B, 1, M, K))