github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 23:38:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5321 from FrankLeeeee/hotfix/accelerator-api

Browse Source 
[accelerator] fixed npu api
This commit is contained in:
Frank Lee 2024-01-29 14:29:58 +08:00 committed by GitHub
commit a6709afe66
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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:
""" """