From 0b47feb785c884eb87831ae8905eed08ec9e936a Mon Sep 17 00:00:00 2001 From: Zhenhuan Chen Date: Wed, 27 Aug 2025 11:58:33 +0800 Subject: [PATCH] [TRTLLM-6960][fix] replace flasky scaled_mm test with more stable config (#7089) Signed-off-by: Zhenhuan Chen --- tests/unittest/_torch/thop/test_scaled_mm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unittest/_torch/thop/test_scaled_mm.py b/tests/unittest/_torch/thop/test_scaled_mm.py index f3cb8ea856c0..32dce08a958f 100644 --- a/tests/unittest/_torch/thop/test_scaled_mm.py +++ b/tests/unittest/_torch/thop/test_scaled_mm.py @@ -31,7 +31,7 @@ ) @pytest.mark.parametrize( "m", - [2048, 12, 228], + [2048, 8, 228], ) @pytest.mark.parametrize( "output_dtype",