Bladeren bron

mpi tests: fix acquire

Samuel Thibault 4 jaren geleden
bovenliggende
commit
f653e1b44d
2 gewijzigde bestanden met toevoegingen van 2 en 2 verwijderingen
  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)
 	{