aboutsummaryrefslogtreecommitdiff
path: root/llama.cpp
diff options
context:
space:
mode:
authorRinne <AsakusaRinne@gmail.com>2023-07-19 15:06:40 +0800
committerGitHub <noreply@github.com>2023-07-19 10:06:40 +0300
commit294f424554c1599784ac9962462fc39ace92d8a5 (patch)
treed70556c6f0a207a1f911c36b37b9b22dcb3065b1 /llama.cpp
parent45a1b07e9b20c33d71d8c849ff27d693a75a0269 (diff)
llama : extend API to get max devices at runtime (#2253)
Diffstat (limited to 'llama.cpp')
-rw-r--r--llama.cpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/llama.cpp b/llama.cpp
index fa3b7c0..3319b70 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -875,6 +875,10 @@ struct llama_model_quantize_params llama_model_quantize_default_params() {
return result;
}
+int llama_max_devices() {
+ return LLAMA_MAX_DEVICES;
+}
+
bool llama_mmap_supported() {
return llama_mmap::SUPPORTED;
}