Merge branch 'testgpuerror' into metal-coreml

ChinChangYang · Dec 11, 2023 · ff76c07 · ff76c07
2 parents 3514467 + bf816e7
commit ff76c07
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cpp/command/gputest.cpp b/cpp/command/gputest.cpp
@@ -78,7 +78,13 @@ int MainCmds::testgpuerror(const vector<string>& args) {
     logger.write("For batch test, using default batch size 16");
   }
   const int maxConcurrentEvals = maxBatchSize * 2 + 16;
-  const int expectedConcurrentEvals = maxBatchSize * 2 + 16;
+  int expectedConcurrentEvals = maxBatchSize * 2 + 16;
+
+#ifdef USE_EIGEN_BACKEND
+  if(expectedConcurrentEvals > 2)
+    expectedConcurrentEvals = 2;
+#endif
+
   const bool defaultRequireExactNNLen = false;
 
   NNEvaluator* nnEval;