CS348Project/llama.cpp
5
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-10-29 08:41:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the validation of main device (#1872)

Browse Source
This commit is contained in:
Howard Su
2023-06-16 01:29:59 +08:00
committed by GitHub
parent 4bfcc855ab
commit 64cc19b4fe
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
ggml-cuda.cu
View File

@@ -2366,7 +2366,7 @@ void ggml_cuda_assign_buffers_no_scratch(struct ggml_tensor * tensor) {
}
void ggml_cuda_set_main_device(int main_device) {
if (main_device > g_device_count) {
if (main_device >= g_device_count) {
fprintf(stderr, "warning: cannot set main_device=%d because there are only %d devices. Using device %d instead.\n",
main_device, g_device_count, g_main_device);
return;