support iluvatar device type

scripts/jiuge.py

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

scripts/libinfinicore_infer.py

@@ -33,6 +33,7 @@ class DeviceType(ctypes.c_int):
     DEVICE_TYPE_ASCEND = 3
     DEVICE_TYPE_METAX = 4
     DEVICE_TYPE_MOORE = 5
+    DEVICE_TYPE_ILUVATAR = 6
 
 
 class JiugeMetaCStruct(ctypes.Structure):