Commit 573ce48
Skip zero-sized parameters in HP fragment mapping (#8298)
## Summary
- exclude zero-sized parameters from ZeRO-1/2 HP-fragment overlap
mapping
- avoid constructing an empty fragment when an empty parameter's offset
lies inside a partition
Fixes #8297.
## Testing
- current `master` (`edaa722`), 2x A6000: minimized ZeRO-1 reproducer
fails before the change with `fragment start 8 should be < fragment_end
8` and completes after the change
- 2-process CPU/Gloo: minimized reproducer completes with both ZeRO-1
and ZeRO-2 after the change
- the trailing zero-sized-parameter reproducer from #8279 still
completes
- `python3 -m py_compile deepspeed/utils/mixed_precision_linkage.py`
Signed-off-by: Cao Yuhang <caoyuhang@fwerkor.com>
Co-authored-by: Olatunji Ruwase <tunji.ruwase@snowflake.com>1 parent f76ab88 commit 573ce48
1 file changed
Lines changed: 5 additions & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
41 | 41 | | |
42 | 42 | | |
43 | 43 | | |
44 | | - | |
45 | | - | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
46 | 47 | | |
47 | 48 | | |
48 | 49 | | |
| 50 | + | |
| 51 | + | |
49 | 52 | | |
50 | 53 | | |
51 | 54 | | |
0 commit comments