vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

bring back android part

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-09-06 17:58:21 +02:00
parent c4a1276ba7
commit 89e70fe3ae
1 changed files with 8 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

9
ggml/src/ggml.c
View file

@ -19556,7 +19556,7 @@ static bool ggml_thread_apply_priority(int32_t prio) {
return true; return true;
} }
#elif (defined(__gnu_linux__) || defined(__FreeBSD__) || defined(__NetBSD__) || defined(__OpenBSD__)) #elif defined(__gnu_linux__)
// TODO: this may not work on BSD, to be verified // TODO: this may not work on BSD, to be verified
static bool ggml_thread_apply_affinity(const bool * mask) { static bool ggml_thread_apply_affinity(const bool * mask) {
@ -19572,7 +19572,14 @@ static bool ggml_thread_apply_affinity(const bool * mask) {
} }
} }
#ifdef __ANDROID__
err = sched_setaffinity(0, sizeof(cpuset), &cpuset);
if (err < 0) {
err = errno;
}
#else
err = pthread_setaffinity_np(pthread_self(), sizeof(cpuset), &cpuset); err = pthread_setaffinity_np(pthread_self(), sizeof(cpuset), &cpuset);
#endif
if (err != 0) { if (err != 0) {
fprintf(stderr, "warn: failed to set affinity mask 0x%llx : %s (%d)\n", (unsigned long long)mask, strerror(err), err); fprintf(stderr, "warn: failed to set affinity mask 0x%llx : %s (%d)\n", (unsigned long long)mask, strerror(err), err);
return false; return false;