Skip to content

Commit 1cc9ae2

Browse files
committed
add dbg
1 parent f7eb848 commit 1cc9ae2

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

comfy/int8_kernels.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -560,6 +560,9 @@ def int8_addmm(
560560
# b has shape [N, K], extract N from first dimension
561561
N = b.shape[0]
562562

563+
# DEBUG: Print shapes to verify
564+
print(f"[DEBUG int8_addmm] a.shape={a.shape}, a_s.shape={a_s.shape}, b.shape={b.shape}, b_s.shape={b_s.shape}, M={M}, N={N}, K={K}")
565+
563566
# Validate shapes
564567
assert b.size(1) == K, f"Shape mismatch: b.shape={b.shape}, expected [..., {K}]"
565568

0 commit comments

Comments
 (0)