ggml/sched : do not skip views in pre-assignments

This commit is contained in:
slaren 2024-11-20 13:25:08 +01:00 committed by Georgi Gerganov
parent 02e4eaf22f
commit 59b9172822
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735

View File

@ -886,9 +886,6 @@ static void ggml_backend_sched_split_graph(ggml_backend_sched_t sched, struct gg
for (int i = 0; i < graph->n_nodes; i++) {
struct ggml_tensor * node = graph->nodes[i];
int * node_backend_id = &tensor_backend_id(node);
if (ggml_is_view_op(node->op)) {
continue;
}
// do not overwrite user assignments
if (*node_backend_id == -1) {
*node_backend_id = ggml_backend_sched_backend_id_from_cur(sched, node);