From 1f13b2435464b97845a6dfc1fd3abb2c49da053a Mon Sep 17 00:00:00 2001 From: biondizzle Date: Tue, 12 May 2026 14:11:53 +0000 Subject: [PATCH] debug: add strides to SF debug prints --- deep_gemm/mega/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_gemm/mega/__init__.py b/deep_gemm/mega/__init__.py index 5a93415..206c6cb 100644 --- a/deep_gemm/mega/__init__.py +++ b/deep_gemm/mega/__init__.py @@ -336,7 +336,7 @@ def fp8_nvfp4_mega_moe(y: torch.Tensor, for name, t in [("l1_w", l1_w), ("l1_w_sf", l1_w_sf), ("l2_w", l2_w), ("l2_w_sf", l2_w_sf)]: - print(f"[debug] {name}: dtype={t.dtype} shape={tuple(t.shape)} contig={t.is_contiguous()}", flush=True) + print(f"[debug] {name}: dtype={t.dtype} shape={tuple(t.shape)} strides={t.stride()} contig={t.is_contiguous()}", flush=True) for name, t in [("sym_x", sym_buffer.x), ("sym_x_sf", sym_buffer.x_sf), ("sym_l1_acts", sym_buffer.l1_acts), ("sym_l1_acts_sf", sym_buffer.l1_acts_sf),