Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Feature] Support engine with NPU backend. #572

Merged
merged 20 commits into from
Oct 24, 2022
Merged

[Feature] Support engine with NPU backend. #572

Changes from 1 commit
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
c8679b8
init npu
wangjiangben-hw Oct 11, 2022
0affc39
Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
wangjiangben-hw Oct 12, 2022
2d57401
Update mmengine/dist/dist.py
wangjiangben-hw Oct 12, 2022
296e331
change to is_hccl_backend
wangjiangben-hw Oct 12, 2022
f78c0dd
Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
wangjiangben-hw Oct 12, 2022
10d2655
add comment with AmpOptimWrapper
wangjiangben-hw Oct 12, 2022
92ac6a7
Merge https://hub.yzuu.cf/wangjiangben-hw/mmengine
wangjiangben-hw Oct 12, 2022
980dbd7
Update mmengine/runner/amp.py
wangjiangben-hw Oct 12, 2022
9787ee2
Update mmengine/runner/amp.py
wangjiangben-hw Oct 12, 2022
2a9d983
add npu fn in base_model
wangjiangben-hw Oct 12, 2022
0b0a212
Merge branch 'main' of https://github.com/wangjiangben-hw/mmengine
wangjiangben-hw Oct 13, 2022
dfd83bf
Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
wangjiangben-hw Oct 13, 2022
9207e0c
clean lint
wangjiangben-hw Oct 13, 2022
5b23b30
Update mmengine/optim/optimizer/amp_optimizer_wrapper.py
wangjiangben-hw Oct 13, 2022
2138438
Update mmengine/model/base_model/base_model.py
wangjiangben-hw Oct 17, 2022
64d6d39
add is_npu_available
wangjiangben-hw Oct 17, 2022
de68262
try to fix
HAOCHENYE Oct 17, 2022
40ebd52
Add comments
HAOCHENYE Oct 17, 2022
915d00c
Refine grammar
HAOCHENYE Oct 17, 2022
6b963f0
Merge commit 'refs/pull/622/head' of https://github.com/open-mmlab/mm…
wangjiangben-hw Oct 19, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update mmengine/runner/amp.py 
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
  • Loading branch information
@wangjiangben-hw @zhouzaida
wangjiangben-hw and zhouzaida authored Oct 12, 2022
commit 9787ee22aec3f102534ea170efb498973c6cfbec
2 changes: 1 addition & 1 deletion mmengine/runner/amp.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ def autocast(device_type: Optional[str] = None,
if is_npu_available():
with torch.npu.amp.autocast(enabled=enabled):
yield
elif torch.cuda.is_available():
elif is_cuda_available():
with torch.cuda.amp.autocast(enabled=enabled):
yield
else:
Expand Down