[NPU] Add NPU support for unit test #4569
tjruwase
merged
21 commits into
deepspeedai:master
from
RUAN-ZX:master
feat: Add npu support for tests/unit/common.py
c3979f68
feat: Add npu support for skip_on_cuda in tests/unit/util.py
cf7e7907
feat: Add npu support for skip_on_arch in tests/unit/util.py
84eec54d
Merge remote-tracking branch 'upstream/master'
2d09de6a
RUAN-ZX
changed the title
RUAN-ZX
changed the title
feat: Set_device of accelerator before deepspeed.init_distributed
9bf85a50
RUAN-ZX
changed the title
Merge branch 'master' into master
9fcda4ac
tjruwase
approved these changes
on 2023-11-01
fix: solve the problem of calling get_accelerator().random().fork_rng…
d650826d
Merge remote-tracking branch 'origin/master'
95537b16
Merge branch 'master' into master
02207091
Merge branch 'master' into master
1d5db419
Merge branch 'master' into master
053b3b0e
loadams
commented
on 2023-11-08
Merge remote-tracking branch 'upstream/master'
795ec248
fix: merge logic branch of xpu and npu
4e0b9aa9
Merge remote-tracking branch 'origin/master'
68c54b28
tjruwase
commented
on 2023-11-09
tjruwase
commented
on 2023-11-09
feat: to check if current accelerator is supported
ee399db7
feat: assert if current accelerator is supported
a9bfab83
feat: add npu support for bf16_required_version_check
b9d20d94
Merge branch 'master' into master
83ba10f3
fix: solve pre-commit hooks problems
ace8fbe1
Merge remote-tracking branch 'upstream/master'
67df1b7f
Merge remote-tracking branch 'origin/master'
1f114b64
tjruwase
commented
on 2023-11-13
tjruwase
approved these changes
on 2023-11-13
Login to write a write a comment.
Login via GitHub