OpenI
/
MegEngine
Not watched
1
0
Fork 0
Code
Releases 31 Wiki Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain
Browse Source

feat(lite): lite support ax compnode 

GitOrigin-RevId: 658ce72d59
tags/v1.7.2.m1
Megvii Engine Team 3 years ago
parent
2ce050bb48
commit
e2427f30e8
2 changed files with 2 additions and 2 deletions
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. +1
    -1
      lite/include/lite/common_enum_c.h
  2. +1
    -1
      lite/pylite/megenginelite/struct.py

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

@@ -34,7 +34,7 @@ typedef enum {
LITE_CAMBRICON = 5, LITE_CAMBRICON = 5,
//! when the device information is set in model, so set LITE_DEVICE_DEFAULT //! when the device information is set in model, so set LITE_DEVICE_DEFAULT
//! in lite, which equal to xpu in megengine //! in lite, which equal to xpu in megengine
LITE_DEVICE_DEFAULT = 6,
LITE_DEVICE_DEFAULT = 7,
} LiteDeviceType; } LiteDeviceType;


typedef enum { typedef enum {


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

@@ -22,7 +22,7 @@ class LiteDeviceType(IntEnum):
LITE_ATLAS = 3 LITE_ATLAS = 3
LITE_NPU = 4 LITE_NPU = 4
LITE_CAMBRICON = 5 LITE_CAMBRICON = 5
LITE_DEVICE_DEFAULT = 6
LITE_DEVICE_DEFAULT = 7




class LiteDataType(IntEnum): class LiteDataType(IntEnum):


Write Preview
Loading…
Cancel
Save