a minor change in function AMPI_GPU_invoke prototype.
[charm.git] / src / libs / ck-libs / ampi / ampi.C
index d27f99721444c7994bd811861c35966e03c83781..3334c69dc989cdf1fe355323aa365148c4e50be8 100644 (file)
@@ -5748,7 +5748,7 @@ int GPUReq::wait(MPI_Status *sts)
 {
     (void)sts;
     while (!isComplete) {
-       AMPI_Suspend(comm);
+       getAmpiInstance(comm)->block();
     }
     return 0;
 }
@@ -5777,7 +5777,7 @@ void AMPI_GPU_complete(void *request, void* dummy)
 }
 
 CDECL
-int AMPI_GPU_invoke(workRequest *to_call, MPI_Request *request, MPI_Comm comm)
+int AMPI_GPU_invoke(workRequest *to_call, MPI_Comm comm, MPI_Request *request)
 {
     AMPIAPI(__func__);