From 818cd433a2de73ff6b0851b2e74c41f3ec247cb2 Mon Sep 17 00:00:00 2001 From: Adam Mainz Date: Thu, 10 Oct 2024 15:44:39 -0700 Subject: [PATCH] fixing typo in fp8_gemm Summary: put an extra `m` by mistake in one of the configs and it is breaking in OSS Reviewed By: plotfi Differential Revision: D64208508 --- userbenchmark/triton/ci.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/userbenchmark/triton/ci.py b/userbenchmark/triton/ci.py index aea648317..7eb6be4ab 100644 --- a/userbenchmark/triton/ci.py +++ b/userbenchmark/triton/ci.py @@ -25,19 +25,19 @@ "fp8_gemm", "--llama", "--only", - "torch_fp8_gemmm, triton_fp8_gemm", + "torch_fp8_gemm, triton_fp8_gemm", ], "fp8_gemm_blockwise": [ "--op", "fp8_gemm_blockwise", - "--extended-shapes", + "--llama", "--only", "_cutlass, _triton", ], "fp8_gemm_rowwise": [ "--op", "fp8_gemm_rowwise", - "--extended-shapes", + "--llama", "--only", "_cutlass,_triton,_cublass", ],