Diego Devesa commited on
Commit
b85e3c0
·
1 Parent(s): cdf95d3

ggml : allow CUDA graphs when using pipeline parallelism (llama/13814)

Browse files
Files changed (1) hide show
  1. ggml/src/ggml-backend.cpp +3 -0
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) {