fix

microsoft · Mar 22, 2024 · c856e4e · c856e4e
1 parent 2c59b46
commit c856e4e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ark/sched/sched/sched_default.cc b/ark/sched/sched/sched_default.cc
@@ -176,6 +176,7 @@ DefaultScheduler::DefaultScheduler(Model &model, int gpu_id, int rank_,
     int num_sm_calc = gpu_info.num_sm - 1;
 
     heuristic_optimize_model(model, model.impl.get(), gpu_info, num_sm_calc);
+    model.create_nodes();
 }
 
 void DefaultScheduler::schedule() {