Sfoglia il codice sorgente

mpi tests: Add missing data acquires

Samuel Thibault 4 anni fa
parent
commit
b6cede7f6f
3 ha cambiato i file con 6 aggiunte e 0 eliminazioni
  1. 2 0
      mpi/tests/mpi_earlyrecv2.c
  2. 2 0
      mpi/tests/mpi_earlyrecv2_sync.c
  3. 2 0
      mpi/tests/mpi_redux.c

+ 2 - 0
mpi/tests/mpi_earlyrecv2.c

@@ -111,12 +111,14 @@ 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);
 	int *rvalue = (int *)starpu_data_get_local_ptr(handle);
 	if (*rvalue != i*other_rank)
 	{
 		FPRINTF_MPI(stderr, "Incorrect received value: %d != %d\n", *rvalue, i*other_rank);
 		*error = 1;
 	}
+	starpu_data_release(handle);
 }
 
 int exchange_variable(int rank, int detached)

+ 2 - 0
mpi/tests/mpi_earlyrecv2_sync.c

@@ -85,6 +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);
 	int *rvalue = (int *)starpu_data_get_local_ptr(handle);
 	if (*rvalue != i*other_rank)
 	{
@@ -95,6 +96,7 @@ void check_variable(starpu_data_handle_t handle, int i, int rank, int *error)
 	{
 		FPRINTF_MPI(stderr, "Correct received value: %d == %d\n", *rvalue, i*other_rank);
 	}
+	starpu_data_release(handle);
 }
 
 int exchange_variable(int rank)

+ 2 - 0
mpi/tests/mpi_redux.c

@@ -68,8 +68,10 @@ int main(int argc, char **argv)
 
 		for(src=1 ; src<size ; src++)
 		{
+			starpu_data_acquire(handles[src], STARPU_R);
 			void *ptr = starpu_data_get_local_ptr(handles[src]);
 			value += *((int *)ptr);
+			starpu_data_release(handles[src]);
 			starpu_data_unregister(handles[src]);
 		}