DeepSpeed
3c1817f3 - Reland perf fix for nan inf check (#7184)

Commit
267 days ago
Reland perf fix for nan inf check (#7184) replace previous usage with logical ops for nan/inf detect with torch.where --------- Signed-off-by: Nadav Elyahu <nelyahu@habana.ai> Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com> Co-authored-by: Hongwei Chen <33092912+hwchen2017@users.noreply.github.com>
Author
Parents
Loading