From 087d0cb1fc6e5f6b60004d952109ec8ce6e32890 Mon Sep 17 00:00:00 2001
From: FrankLeeeee <somerlee.9@gmail.com>
Date: Mon, 29 Jan 2024 14:27:52 +0800
Subject: [PATCH] [accelerator] fixed npu api

---
 colossalai/accelerator/npu_accelerator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/colossalai/accelerator/npu_accelerator.py b/colossalai/accelerator/npu_accelerator.py
index 1a86f84cb..b28492968 100644
--- 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:
         """