[NVFP4][fix] Fix layer.weight -> w13 typo in NVFP4 MOE emulation kernel preparation (#41630)

Signed-off-by: Felix Marty <[email protected]>
This commit is contained in:
fxmarty-amd
2026-05-04 20:13:37 +00:00
committed by GitHub
parent 844df54269
commit 9c07342fdc
@@ -381,7 +381,7 @@ def convert_to_nvfp4_moe_kernel_format(
elif nvfp4_backend == NvFp4MoeBackend.EMULATION:
# Move the E2M1 lookup table to the device now, because
# `.to(device)` is not allowed during CUDA graph capture.
kE2M1ToFloat_handle.val = kE2M1ToFloat_handle.val.to(layer.weight.device)
kE2M1ToFloat_handle.val = kE2M1ToFloat_handle.val.to(w13.device)
if a13_scale is None or a2_scale is None:
raise ValueError(