PR #5826 Move inf_or_nan_tracker to cpu for cpu offload

Move inf_or_nan_tracker to cpu for cpu offload #5826

loadams merged 6 commits into deepspeedai:master from BacharL:inf_nan_tracker

move inf_or_nan_tracker to cpu for cpu offload

89dc9633

BacharL requested a review from

tjruwase 1 year ago

Merge branch 'master' into inf_nan_tracker

c82abcd3

replace device index with device name

963f21f4

tjruwase approved these changes on 2024-08-07

BacharL force pushed from 998a00b1 to 963f21f4 1 year ago

Merge branch 'master' into inf_nan_tracker

f7d6ee60

formatting fixes

09b592ac

Merge branch 'master' into inf_nan_tracker

c340f6af

loadams enabled auto-merge 1 year ago

loadams merged 4d4ff0ed into master 1 year ago

Reviewers

tjruwase

Assignees

No one assigned

Labels

None yet

Milestone

No milestone