update gpt-2 example

diff --git a/examples/gpt-2/main.cpp b/examples/gpt-2/main.cpp
index a0109ffeeed565f5db7f53417f238bbe09bac953..9d3ae92c167edd56aca8186f5ffe18a5bdb9ccb7 100644 (file)
--- a/examples/gpt-2/main.cpp
+++ b/examples/gpt-2/main.cpp
@@ -947,7 +947,7 @@ int main(int argc, char ** argv) {
     ggml_backend_sched_t sched;
     {
         // initialize the scheduler
-        sched = ggml_backend_sched_new(model.backends.data(), model.backends.size());
+        sched = ggml_backend_sched_new(model.backends.data(), NULL, model.backends.size(), GPT2_MAX_NODES);
 
         // create the worst case graph for memory usage estimation
         int n_tokens = std::min(model.hparams.n_ctx, params.n_batch);