issue/48: p800支持

6fa3a12a · zhangyue · GitHub · 1ab1e668 · 6fa3a12a · 6fa3a12a
Unverified Commit 6fa3a12a authored Sep 18, 2025 by zhangyue Committed by GitHub Sep 18, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

scripts/jiuge.py scripts/jiuge.py +2 -0

scripts/libinfinicore_infer/base.py scripts/libinfinicore_infer/base.py +1 -0

No files found.
--- a/scripts/jiuge.py
+++ b/scripts/jiuge.py
@@ -675,6 +675,8 @@ def test():
        device_type = DeviceType.DEVICE_TYPE_MOORE
    elif sys.argv[1] == "--iluvatar":
        device_type = DeviceType.DEVICE_TYPE_ILUVATAR
+    elif sys.argv[1] == "--kunlun":
+        device_type = DeviceType.DEVICE_TYPE_KUNLUN
    else:
        print(
            "Usage: python jiuge.py [--cpu | --nvidia| --cambricon | --ascend | --metax | --moore] <path/to/model_dir> [n_device]"

--- a/scripts/libinfinicore_infer/base.py
+++ b/scripts/libinfinicore_infer/base.py
@@ -34,6 +34,7 @@ class DeviceType(ctypes.c_int):
    DEVICE_TYPE_METAX = 4
    DEVICE_TYPE_MOORE = 5
    DEVICE_TYPE_ILUVATAR = 6
+    DEVICE_TYPE_KUNLUN = 7


 class KVCacheCStruct(ctypes.Structure):