Spaces:
Running
Running
Diego Devesa commited on
Commit ·
b85e3c0
1
Parent(s): cdf95d3
ggml : allow CUDA graphs when using pipeline parallelism (llama/13814)
Browse files
ggml/src/ggml-backend.cpp
CHANGED
|
@@ -1598,6 +1598,9 @@ void ggml_backend_sched_synchronize(ggml_backend_sched_t sched) {
|
|
| 1598 |
for (int i = 0; i < sched->n_backends; i++) {
|
| 1599 |
ggml_backend_synchronize(sched->backends[i]);
|
| 1600 |
}
|
|
|
|
|
|
|
|
|
|
| 1601 |
}
|
| 1602 |
|
| 1603 |
void ggml_backend_sched_set_eval_callback(ggml_backend_sched_t sched, ggml_backend_sched_eval_callback callback, void * user_data) {
|
|
|
|
| 1598 |
for (int i = 0; i < sched->n_backends; i++) {
|
| 1599 |
ggml_backend_synchronize(sched->backends[i]);
|
| 1600 |
}
|
| 1601 |
+
// reset the current copy to 0 so that the graphs will be similar during generation
|
| 1602 |
+
// necessary for CUDA graphs
|
| 1603 |
+
sched->cur_copy = 0;
|
| 1604 |
}
|
| 1605 |
|
| 1606 |
void ggml_backend_sched_set_eval_callback(ggml_backend_sched_t sched, ggml_backend_sched_eval_callback callback, void * user_data) {
|