EngineX-Ascend/enginex-ascend-910-llama.cpp
10
0
Fork 0
Code Issues Pull Requests Actions 4 Projects Releases Wiki Activity

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

Browse Source 
* 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
1 changed files with 2 additions and 2 deletions
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) {