PR #5125 compute global norm on device

compute global norm on device #5125

loadams merged 7 commits into deepspeedai:master from BacharL:global_norm_on_device

BacharL requested a review from

mrwyattii 1 year ago

BacharL requested a review from

tjruwase 1 year ago

tjruwase commented on 2024-02-13

BacharL force pushed from 0541f4ef to 74f5b701 1 year ago

tjruwase approved these changes on 2024-02-15

BacharL force pushed from 03272aa4 to 53246a71 1 year ago

BacharL force pushed from 92d5dbbe to 283b3b24 1 year ago

BacharL force pushed from 283b3b24 to b30640b9 1 year ago

compute global norm on device

b9c2059e

code review comments

676b1992

code review comments

dc9ed13b

reuse graph code path for norm computation

bf5be3a5

BacharL force pushed from b30640b9 to bf5be3a5 1 year ago

Fix formatting

56f13cef

fix return value device to match input device

ce15b963

Merge branch 'master' into global_norm_on_device

5e4f8905

loadams enabled auto-merge 1 year ago

loadams merged 9f0e2136 into master 1 year ago

Reviewers

tjruwase

mrwyattii

Assignees

No one assigned

Labels

None yet

Milestone

No milestone