DeepSpeed
195563a2 - fix a small type error on bf16+pp (#3441)

Commit

2 years ago

fix a small type error on bf16+pp (#3441) Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>

References

#3441 - Fix a type error on bf16+Pipeline Parallelism

Author

ys950902

Parents

f1fab902

Files1

deepspeed/runtime/pipe
- engine.py

DeepSpeed 195563a2 - fix a small type error on bf16+pp (#3441)

DeepSpeed
195563a2 - fix a small type error on bf16+pp (#3441)