1 files changed, 1 insertions, 1 deletions
diff --git a/llama.cpp b/llama.cpp
index 650e8bba..b19c6ff3 100644
--- a/llama.cpp
+++ b/llama.cpp
@@ -16372,7 +16372,7 @@ struct llama_context * llama_new_context_with_model(
             return nullptr;
         }
         if (model->split_mode == LLAMA_SPLIT_MODE_NONE) {
-            ggml_backend_t backend = ggml_backend_vk_init(0);
+            ggml_backend_t backend = ggml_backend_vk_init(model->main_gpu);
             if (backend == nullptr) {
                 LLAMA_LOG_ERROR("%s: failed to initialize Vulkan backend\n", __func__);
                 llama_free(ctx);