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

[accelerator] fixed npu api
2024-01-29 14:29:58 +08:00 · 2024-01-29 14:29:58 +08:00 · a6709afe66
parent 8823cc4831 087d0cb1fc
commit a6709afe66
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:
        """