Skip to content

[PyTorch] Fix DistributedWeight handling under saved-tensor hooks and in the fused grouped MLP - #3545

Open
xrennvidia wants to merge 3 commits into
NVIDIA:mainfrom
xrennvidia:xren/dist-weight-saved-tensor-hooks
Open

[PyTorch] Fix DistributedWeight handling under saved-tensor hooks and in the fused grouped MLP#3545
xrennvidia wants to merge 3 commits into
NVIDIA:mainfrom
xrennvidia:xren/dist-weight-saved-tensor-hooks

[PyTorch] Drop NVFP4 rowwise input copies from the fused grouped MLP …

250974f
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Oct 18, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO