diff --git a/lite/include/lite/common_enum_c.h b/lite/include/lite/common_enum_c.h index 656f6ee7..97bcd70e 100644 --- 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 { diff --git a/lite/pylite/megenginelite/struct.py b/lite/pylite/megenginelite/struct.py index 937c46c6..0d54f375 100644 --- 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):