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",
         }