Skip to content

Commit 8ce677c

Browse files
Unireversedongchengwei
andauthored
[Fix](mlu-ops): Add MTP617 in core (#1290)
Co-authored-by: dongchengwei <dongchengwei@cambricon.com>
1 parent 554cadd commit 8ce677c

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

core/context.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,10 @@ mluOpDevType_t convertDeviceNameFromInt(int device_code) {
6060
return MLUOP_MTP613;
6161
break;
6262
}
63+
case 617: {
64+
return MLUOP_MTP617;
65+
break;
66+
}
6367
default:
6468
break;
6569
}

core/context.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ typedef enum {
5858
MLUOP_MLU370 = 372,
5959
MLUOP_MLU590 = 592,
6060
MLUOP_MTP613 = 613,
61+
MLUOP_MTP617 = 617,
6162
MLUOP_MLU290 = 290,
6263
} mluOpDevType_t;
6364

0 commit comments

Comments
 (0)