From e2427f30e825f00b9a3fa69c645b596c48b7b7af Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 22 Dec 2021 16:56:03 +0800 Subject: [PATCH] feat(lite): lite support ax compnode GitOrigin-RevId: 658ce72d596f758e0f9b0e9dac70dd420c8742a1 --- lite/include/lite/common_enum_c.h | 2 +- lite/pylite/megenginelite/struct.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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):