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

Loading…
Cancel
Save