Make sure we pick the reduced tensor from the right GPU

This commit is contained in:
Kawrakow
2026-02-23 09:37:34 +00:00
parent 68bd30d99c
commit fd2a70913c
4 changed files with 45 additions and 9 deletions

View File

@@ -186,6 +186,7 @@ extern "C" {
GGML_API int ggml_backend_sched_get_n_backends(ggml_backend_sched_t sched);
GGML_API ggml_backend_t ggml_backend_sched_get_backend(ggml_backend_sched_t sched, int i);
GGML_API int ggml_backend_sched_get_backend_idx(ggml_backend_sched_t sched, ggml_backend_buffer_t buffer);
// Get the number of splits of the last graph
GGML_API int ggml_backend_sched_get_n_splits(ggml_backend_sched_t sched);