浏览代码

bidouille

Romain LION 5 年之前
父节点
当前提交
f5b3eb01ac
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      mpi/src/mpi_failure_tolerance/starpu_mpi_checkpoint.c

+ 2 - 1
mpi/src/mpi_failure_tolerance/starpu_mpi_checkpoint.c

@@ -158,7 +158,8 @@ int starpu_mpi_submit_checkpoint_template(starpu_mpi_checkpoint_template_t cp_te
 	int current_instance;
 
 	current_instance = increment_current_instance();
-	STARPU_ASSERT(current_instance<4);
+	if (current_instance>3)
+		starpu_task_wait_for_all();
 	_starpu_mpi_checkpoint_post_cp_discard_recv(cp_template);
 	_starpu_mpi_checkpoint_template_create_instance_tracker(cp_template, cp_template->cp_id, cp_template->checkpoint_domain, current_instance);
 	//TODO check what happens when all the ack msg are received when we arrrive here.