From 37e1479a70f63a397ec6ee6e94e22398263057bd Mon Sep 17 00:00:00 2001 From: Wei Feng Date: Thu, 26 Sep 2024 11:48:46 -0700 Subject: [PATCH] test CI Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags: --- test/float8/test_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/float8/test_base.py b/test/float8/test_base.py index 60d8ffa57..ba2d49033 100644 --- a/test/float8/test_base.py +++ b/test/float8/test_base.py @@ -600,7 +600,7 @@ def test_small_amax_float16(self, float8_dtype): def test_dynamic_scale_parity(self, dtype: torch.dtype): scaling_type_weight = ScalingType.DYNAMIC torch.manual_seed(0) - hp_tensor = torch.randn(32, 32, device="cuda", dtype=dtype) + hp_tensor = torch.randn(16, 16, device="cuda", dtype=dtype) float8_config = Float8LinearConfig( cast_config_weight=CastConfig(scaling_type=scaling_type_weight), )