deepspeedai/DeepSpeed

Pull Requests Commits

[create-pull-request] automated change

loadams committed 1 day ago

904d018f

Honor adam_w_mode in the CPU multi_tensor_adam binding (#8307)

PKUWZP committed 1 day ago

Verified c7cc64a9

Route ZeRO/SuperOffload pin sites through accelerator pin_memory (#8256)

sfc-gh-truwase committed 1 day ago

Verified eebe24a6

[AutoTP] Fix training lm_head routing (#8302)

gaoxiaomo committed 1 day ago

Verified 1b956e62

Fix ZeRO-3 synchronization during OPSD rollout (#8264)

LiRunGuo committed 1 day ago

Verified e40c681b

Isolate DeepCompile list-schedule test ops (#8319)

tohtana committed 1 day ago

Verified 28ed612e

Fix shared loss gradient accumulation (#8245)

tohtana committed 1 day ago

Verified 4ffb4e5c

stop allocating per-element temporaries in the overflow check (#8325)

stas00 committed 1 day ago

Verified 32e301ff

feat(opsd): share prompt prefill across rollout samples (#8296)

nathon-lee committed 2 days ago

Verified a9505fd6

Enable optimized Adam backend for MuonWithAuxAdam optimizer (#8278)

jinyouzhi committed 2 days ago

Verified d47e4f4d

[Apple Silicon Support Phase 1] Add Metal FusedAdam kernel and CPU Adam build for Apple Silicon (#8300)

PKUWZP committed 2 days ago

Verified 715965e0

Fix AutoTP metadata updates for unsharded modules (#8299)

zupengwang committed 3 days ago

Verified 7154a00f

Avoid redundant copies in MPS P2P staging (#8314)

FU-max-boop committed 3 days ago

Verified eba5d273

Reject invalid ZenFlow ratio and update interval boundaries (#8274)

tandede committed 3 days ago

Verified 6e3bd087

Import print_dist in auto_tp (#8311)

alanhuangyoo committed 3 days ago

Verified b753aec8

Fix device mismatch in test_gate_up_partition_covers_the_whole_weight (#8308)

alanhuangyoo committed 4 days ago

Verified 84fd92aa

Route isend/irecv to nonblocking backend methods and stage them as async on MPS (#8303)

PKUWZP committed 4 days ago

Verified 9311fd55

perf(rollout): add HybridEngine rollout profiling (#8295)

nathon-lee committed 4 days ago

Verified da3ca683

Skip zero-sized parameters in HP fragment mapping (#8298)

fwerkor committed 4 days ago

Verified 573ce488

Drop the documented grad_hooks ZeRO option, which does not exist (#8242)

vineethsaivs committed 4 days ago

Verified f76ab880

Register native pinned host memory with CUDA for GPU DMA (#8283)

sfc-gh-truwase committed 4 days ago

Verified cb260806

Fix zero reduce bucket size validation (#8266)

fwerkor committed 5 days ago

Verified e70b9035

[AutoTP] Convert HF embedding_rowwise tp_plan entries to SKIP specs (#8294)

ebarkhordar committed 5 days ago

Verified 20ea4543

fix(engine): resolve inference workspace attribute lookup (#8288)

nathon-lee committed 5 days ago

Verified fb57b819

Enable DeepSpeed support on Apple Silicon (MPS) with ZeRO Stage 1-3 (#8293)

PKUWZP committed 5 days ago

Verified 64fcec6b

Fix repeated FlopsProfiler metric accumulation (#8246)

baremetaldevx86 committed 5 days ago

Verified ebb75d7e

fix: stop DeepSpeedConfig writing max_grad_norm back into the caller's config dict (#8289)

ebarkhordar committed 5 days ago

Verified f5671753

Fix ZeroDivisionError in compute_elastic_config return_microbatch on non-0.2 elasticity (#8286)

ebarkhordar committed 5 days ago

Verified ad1c516e

docs: note async cpu_checkpointing perf and expandable_segments (#8287)

sfc-gh-truwase committed 5 days ago

Verified 96ffef2c

Fix ZeRO-1/2 with zero-sized parameters (#8280)

fwerkor committed 6 days ago

Verified edaa7221

Older