added explicit branch for auto core mask, changed default case to fail

PhotonVision · Jan 29, 2024 · 7045c21 · 7045c21
1 parent ecc58ce
commit 7045c21
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/native/cpp/yolo_common.cpp b/src/main/native/cpp/yolo_common.cpp
@@ -86,6 +86,9 @@ YoloModel::YoloModel(std::string modelPath, int num_classes_, ModelVersion type_
     rknn_core_mask core_mask;
     switch (coreNumber)
     {
+    case -1:
+        core_mask = RKNN_NPU_CORE_AUTO;
+        break;
     case 0:
         core_mask = RKNN_NPU_CORE_0;
         break;
@@ -102,7 +105,7 @@ YoloModel::YoloModel(std::string modelPath, int num_classes_, ModelVersion type_
         core_mask = RKNN_NPU_CORE_0_1_2;
         break;
     default:
-        core_mask = RKNN_NPU_CORE_AUTO;
+        throw std::runtime_error("invalid core selection! core selected: " + coreNumber);
         break;
     }
     ret = rknn_set_core_mask(ctx, core_mask);