Skip to content

Commit a3d50bc

Browse files
authored
ggml-backend : only offload from host buffers (ggml-org#11120)
1 parent a4dd490 commit a3d50bc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-backend.cpp

+1-1
Original file line numberDiff line numberDiff line change
@@ -761,7 +761,7 @@ static int ggml_backend_sched_backend_id_from_cur(ggml_backend_sched_t sched, st
761761
}
762762
// skip ROPE since the rope freqs tensor is too small to choose a backend based on it
763763
// not an ideal solution
764-
if (tensor->op != GGML_OP_ROPE && src->buffer != NULL && src->buffer->usage == GGML_BACKEND_BUFFER_USAGE_WEIGHTS) {
764+
if (tensor->op != GGML_OP_ROPE && src->buffer != NULL && src->buffer->usage == GGML_BACKEND_BUFFER_USAGE_WEIGHTS && ggml_backend_buffer_is_host(src->buffer)) {
765765
int src_backend_id = ggml_backend_sched_backend_from_buffer(sched, src, tensor);
766766
// check if a backend with higher prio wants to offload the op
767767
if (src_backend_id == sched->n_backends - 1) {

0 commit comments

Comments
 (0)