Browse Source

test-barrier : do not use more threads than physically available (#16389)

* do not use more threads than physically available

* ensure n_threads > 0

Co-authored-by: Jeff Bolz <jbolz@nvidia.com>

---------

Co-authored-by: Jeff Bolz <jbolz@nvidia.com>
Sigbjørn Skjæret 3 tháng trước cách đây
mục cha
commit
d64c8104f0
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      tests/test-barrier.cpp

+ 2 - 2
tests/test-barrier.cpp

@@ -1,6 +1,5 @@
 #include "ggml.h"
 #include "ggml-cpu.h"
-#include "ggml-backend.h"
 
 #include <chrono>
 #include <iostream>
@@ -8,12 +7,13 @@
 #include <cstdlib>
 #include <cassert>
 #include <vector>
+#include <thread>
 
 #define MAX_NARGS 2
 
 int main(int argc, char *argv[]) {
 
-    int n_threads = 4;
+    int n_threads = std::max(1, std::min(4, (int) std::thread::hardware_concurrency()));
     int n_rounds  = 100;
 
     if (argc > 1) {