Skip to content

Commit 99052cd

Browse files
authored
sched : offload_op also requires supports_op (ggml-org#7977)
1 parent c637fcd commit 99052cd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml-backend.c

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1172,7 +1172,7 @@ static int ggml_backend_sched_backend_id_from_cur(ggml_backend_sched_t sched, st
11721172
// check if a backend with higher prio wants to offload the op
11731173
if (src_backend_id == sched->n_backends - 1) {
11741174
for (int b = 0; b < src_backend_id; b++) {
1175-
if (ggml_backend_offload_op(sched->backends[b], tensor)) {
1175+
if (ggml_backend_supports_op(sched->backends[b], tensor) && ggml_backend_offload_op(sched->backends[b], tensor)) {
11761176
SET_CAUSE(tensor, "1.off");
11771177
return b;
11781178
}

0 commit comments

Comments
 (0)