Update

[ghstack-poisoned]
pytorch · Jan 8, 2025 · 42e0a07 · 42e0a07
1 parent cf14e55
commit 42e0a07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/float8/profile_linear_float8.py b/benchmarks/float8/profile_linear_float8.py
@@ -355,7 +355,7 @@ def main(
             1, 2048, 4096, device=device, dtype=ref_dtype
         ).requires_grad_()
     else:
-        M, K, N = 4096, 4096, 4096
+        M, K, N = 2048, 4096, 8192
         m_ref = torch.nn.Sequential(
             torch.nn.Linear(K, N, bias=False),
         )