Browse Source

[accelerator] fixed npu api

pull/5321/head
FrankLeeeee 10 months ago
parent
commit
087d0cb1fc
  1. 2
      colossalai/accelerator/npu_accelerator.py

2
colossalai/accelerator/npu_accelerator.py

@ -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:
"""

Loading…
Cancel
Save