test CI

Summary: Test Plan: Reviewers: Subscribers: Tasks: Tags:
pytorch · Sep 26, 2024 · 37e1479 · 37e1479
1 parent 5900c3e
commit 37e1479
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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),
         )