happyz
/
llama.cpp
mirror of https://github.com/ggerganov/llama.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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>
This commit is contained in:
Sigbjørn Skjæret 2025-10-02 20:10:12 +02:00 committed by GitHub
parent ef07a40906
commit d64c8104f0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test-barrier.cpp
View File

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