support ascend 910B2 (#885)

Signed-off-by: ouyangluwei <[email protected]>
Co-authored-by: ouyangluwei <[email protected]>

Author: ouyangluwei163

charts/hami/templates/scheduler/device-configmap.yaml

@@ -119,6 +119,27 @@ data:
         - name: vir16
           memory: 17476
           aiCore: 16
+    - chipName: 910B2
+      commonWord: Ascend910B2
+      resourceName: huawei.com/Ascend910B2
+      resourceMemoryName: huawei.com/Ascend910B2-memory
+      memoryAllocatable: 65536
+      memoryCapacity: 65536
+      aiCore: 24
+      aiCPU: 6
+      templates:
+        - name: vir03_1c_8g
+          memory: 8192
+          aiCore: 3
+          aiCPU: 1
+        - name: vir06_1c_16g
+          memory: 16384
+          aiCore: 6
+          aiCPU: 1
+        - name: vir12_3c_32g
+          memory: 32768
+          aiCore: 12
+          aiCPU: 3
     - chipName: 910B3
       commonWord: Ascend910B
       resourceName: huawei.com/Ascend910B

charts/hami/values.yaml

@@ -200,6 +200,8 @@ devices:
     customresources:
       - huawei.com/Ascend910A
       - huawei.com/Ascend910A-memory
+      - huawei.com/Ascend910B2
+      - huawei.com/Ascend910B2-memory
       - huawei.com/Ascend910B
       - huawei.com/Ascend910B-memory
       - huawei.com/Ascend910B4

pkg/device/devices.go

@@ -249,6 +249,27 @@ vnpus:
       - name: "vir16"
         memory: 17476
         aiCore: 16
+  - chipName: 910B2
+    commonWord: Ascend910B2
+    resourceName: huawei.com/Ascend910B2
+    resourceMemoryName: huawei.com/Ascend910B2-memory
+    memoryAllocatable: 65536
+    memoryCapacity: 65536
+    aiCore: 24
+    aiCPU: 6
+    templates:
+      - name: vir03_1c_8g
+        memory: 8192
+        aiCore: 3
+        aiCPU: 1
+      - name: vir06_1c_16g
+        memory: 16384
+        aiCore: 6
+        aiCPU: 1
+      - name: vir12_3c_32g
+        memory: 32768
+        aiCore: 12
+        aiCPU: 3
   - chipName: "910B3"
     commonWord: "Ascend910B"
     resourceName: "huawei.com/Ascend910B"

pkg/device/devices_test.go

@@ -89,6 +89,27 @@ vnpus:
     - name: vir16
       memory: 17476
       aiCore: 16
+- chipName: 910B2
+  commonWord: Ascend910B2
+  resourceName: huawei.com/Ascend910B2
+  resourceMemoryName: huawei.com/Ascend910B2-memory
+  memoryAllocatable: 65536
+  memoryCapacity: 65536
+  aiCore: 24
+  aiCPU: 6
+  templates:
+    - name: vir03_1c_8g
+      memory: 8192
+      aiCore: 3
+      aiCPU: 1
+    - name: vir06_1c_16g
+      memory: 16384
+      aiCore: 6
+      aiCPU: 1
+    - name: vir12_3c_32g
+      memory: 32768
+      aiCore: 12
+      aiCPU: 3
 - chipName: 910B3
   commonWord: Ascend910B
   resourceName: huawei.com/Ascend910B