feat(lite): lite support ax compnode

GitOrigin-RevId: 658ce72d596f758e0f9b0e9dac70dd420c8742a1

feat(lite): lite support ax compnode
GitOrigin-RevId: 658ce72d596f758e0f9b0e9dac70dd420c8742a1
e2427f30 · Megvii Engine Team · 2ce050bb · e2427f30 · e2427f30
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

lite/include/lite/common_enum_c.h lite/include/lite/common_enum_c.h +1 -1

lite/pylite/megenginelite/struct.py lite/pylite/megenginelite/struct.py +1 -1

未找到文件。
--- a/lite/include/lite/common_enum_c.h
+++ b/lite/include/lite/common_enum_c.h
@@ -34,7 +34,7 @@ typedef enum {
    LITE_CAMBRICON = 5,
    //! when the device information is set in model, so set LITE_DEVICE_DEFAULT
    //! in lite, which equal to xpu in megengine
-    LITE_DEVICE_DEFAULT = 6,
+    LITE_DEVICE_DEFAULT = 7,
 } LiteDeviceType;
 typedef enum {

--- a/lite/pylite/megenginelite/struct.py
+++ b/lite/pylite/megenginelite/struct.py
@@ -22,7 +22,7 @@ class LiteDeviceType(IntEnum):
    LITE_ATLAS = 3
    LITE_NPU = 4
    LITE_CAMBRICON = 5
-    LITE_DEVICE_DEFAULT = 6
+    LITE_DEVICE_DEFAULT = 7
 class LiteDataType(IntEnum):