Skip to content

test: guard against per-chunk lm_head all-gather in chunked_nll under…#6172

Draft
behroozazarkhalili wants to merge 2 commits into
mainfrom
test/chunked-nll-fsdp2-allgather-guard
Draft

test: guard against per-chunk lm_head all-gather in chunked_nll under…#6172
behroozazarkhalili wants to merge 2 commits into
mainfrom
test/chunked-nll-fsdp2-allgather-guard