github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-22 05:29:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

[accelerator] fixed npu api

Browse Source
This commit is contained in:
FrankLeeeee 2024-01-29 14:27:52 +08:00
parent 8823cc4831
commit 087d0cb1fc
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/accelerator/npu_accelerator.py
View File

@ -31,7 +31,7 @@ class NpuAccelerator(BaseAccelerator):
"""
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:
"""