Add NPU FusedAdam support #4343
tjruwase
merged
25 commits into
deepspeedai:master
from
CurryRice233:master
add npu support dtypes
e37215d4
Merge remote-tracking branch 'origin/master'
8614f404
add npu fused_adam support
ac78f3a7
Merge branch 'master' into master
fd646193
Merge branch 'master' into master
a43fcf8b
Merge branch 'master' into master
06172068
add license
c87ef320
Merge remote-tracking branch 'origin/master'
5eaeee07
Merge branch 'master' into master
510166e3
Merge branch 'master' into master
efe4c128
tjruwase
approved these changes
on 2023-10-06
Merge branch 'master' into master
51ddc02b
Merge branch 'master' into master
b472ecf9
Update accelerator/npu_accelerator.py
258eac1b
Update op_builder/npu/fused_adam.py
620b85e7
Update op_builder/npu/fused_adam.py
dd94fcc4
Update op_builder/npu/fused_adam.py
b41f20cf
Update op_builder/npu/fused_adam.py
0487035c
Update op_builder/npu/fused_adam.py
52a2a8c6
Update op_builder/npu/fused_adam.py
d808f862
Update op_builder/npu/fused_adam.py
fb77cd05
Merge branch 'master' into master
38bc9268
Update accelerator/npu_accelerator.py
7b26ce40
Update accelerator/npu_accelerator.py
da7edd65
Merge branch 'master' into master
4b6197fe
Merge branch 'master' into master
20bd85eb
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub