haraldwolff/llama.cpp
1
0
Fork 0
Code Issues Pull requests 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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats 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;