Repository navigation

Count ep and fsdp in caching_allocator_warmup byte calculation - #49384

Merged

3outeille merged 6 commits into

mainfrom

fix-warmup-byte-count-ep-plan

Oct 12, 2026

Merged

Count ep and fsdp in caching_allocator_warmup byte calculation#49384
3outeille merged 6 commits into
mainfrom
fix-warmup-byte-count-ep-plan

Commits on Oct 7, 2026
- [distributed] Count EP-sharded experts in the allocator warmup, and keep the model's plans resolved
  
  3outeille
  authored and
  ferdinand_mom
  committed
  87f176f
Commits on Oct 8, 2026
- use dtensor instead
  3outeille
  committed
  3eb6382
- Merge branch 'main' into fix-warmup-byte-count-ep-plan
  3outeille
  authored
  a72a390
- Merge branch 'main' into fix-warmup-byte-count-ep-plan
  3outeille
  authored
  09d99e3
- clean
  3outeille
  committed
  068aadd
- Merge branch 'fix-warmup-byte-count-ep-plan' of github.com:huggingface/transformers into fix-warmup-byte-count-ep-plan
  3outeille
  committed
  d25897f