From a98c9be7e79bcb8ac62d193e1ab7322f72b4bf22 Mon Sep 17 00:00:00 2001
From: wooway777 <wooway777@gmail.com>
Date: Tue, 7 Jul 2026 13:17:20 +0800
Subject: [PATCH] fix: update kunlunxin torch type

---
 python/infinilm/base_config.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/infinilm/base_config.py b/python/infinilm/base_config.py
index 4023e603..f19f3508 100644
--- a/python/infinilm/base_config.py
+++ b/python/infinilm/base_config.py
@@ -505,7 +505,7 @@ def get_device_str(self, device):
             "metax": "cuda",
             "moore": "musa",
             "iluvatar": "cuda",
-            "kunlun": "kunlun",
+            "kunlun": "cuda",
             "hygon": "cuda",
             "ali": "cuda",
         }