Skip to content

Commit

Permalink
Merge pull request #5321 from FrankLeeeee/hotfix/accelerator-api
Browse files Browse the repository at this point in the history
[accelerator] fixed npu api
  • Loading branch information
FrankLeeeee authored Jan 29, 2024
2 parents 8823cc4 + 087d0cb commit a6709af
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion colossalai/accelerator/npu_accelerator.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ def get_version(self) -> str:
"""
Return the version of the accelerator which torch is built against.
"""
return torch.version.npu
return torch.version.cann

def get_current_device(self) -> torch.device:
"""
Expand Down

0 comments on commit a6709af

Please sign in to comment.