github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[accelerator] fixed npu api

pull/5321/head
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 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: def get_current_device(self) -> torch.device:
""" """