[accelerator] fixed npu api

2024-01-29 14:27:52 +08:00 · 2024-01-29 14:27:52 +08:00 · 087d0cb1fc
parent 8823cc4831
commit 087d0cb1fc
1 changed files with 1 additions and 1 deletions
--- a/colossalai/accelerator/npu_accelerator.py
+++ b/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:
        """