Skip to content

[PyTorch] Fix MXFP8 master weight cast when a rank owns no shard. - #3348

Open
rapatel wants to merge 3 commits into
NVIDIA:mainfrom
rapatel:ripatel/handle_empty_shards
Open

[PyTorch] Fix MXFP8 master weight cast when a rank owns no shard.#3348
rapatel wants to merge 3 commits into
NVIDIA:mainfrom
rapatel:ripatel/handle_empty_shards

[PyTorch] Add a 2-rank MXFP8 empty-shard test where one rank owns the…

554ca0a
Select commit
Loading
Failed to load commit list.
Sign in for the full log view