Skip to content

Commit 027d97e

Browse files
WIP
1 parent ea3cab5 commit 027d97e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/llama-context.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -221,7 +221,7 @@ llama_context::llama_context(
221221
bool pipeline_parallel =
222222
model.n_devices() > 1 &&
223223
model.params.n_gpu_layers > (int) model.hparams.n_layer &&
224-
model.params.split_mode == LLAMA_SPLIT_MODE_LAYER &&
224+
(model.params.split_mode == LLAMA_SPLIT_MODE_LAYER || model.params.split_mode == LLAMA_SPLIT_MODE_ROW) &&
225225
cparams.offload_kqv &&
226226
!model.has_tensor_overrides();
227227

0 commit comments

Comments
 (0)