浏览代码

mpi tests: fix acquire

Samuel Thibault 4 年之前
父节点
当前提交
f653e1b44d
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 1 1
      mpi/tests/mpi_earlyrecv2.c
  2. 1 1
      mpi/tests/mpi_earlyrecv2_sync.c

+ 1 - 1
mpi/tests/mpi_earlyrecv2.c

@@ -111,7 +111,7 @@ void check_variable(starpu_data_handle_t handle, int i, int rank, int *error)
 {
 	int other_rank = rank%2 == 0 ? rank+1 : rank-1;
 
-	starpu_data_acquire(handle, STARPU_R);
+	starpu_data_acquire_on_node(handle, starpu_worker_get_local_memory_node(), STARPU_R);
 	int *rvalue = (int *)starpu_data_get_local_ptr(handle);
 	if (*rvalue != i*other_rank)
 	{

+ 1 - 1
mpi/tests/mpi_earlyrecv2_sync.c

@@ -85,7 +85,7 @@ void check_variable(starpu_data_handle_t handle, int i, int rank, int *error)
 {
 	int other_rank = rank%2 == 0 ? rank+1 : rank-1;
 
-	starpu_data_acquire(handle, STARPU_R);
+	starpu_data_acquire_on_node(handle, starpu_worker_get_local_memory_node(), STARPU_R);
 	int *rvalue = (int *)starpu_data_get_local_ptr(handle);
 	if (*rvalue != i*other_rank)
 	{