OSC UCX: Setting the lock type when MPI_MODE_NOCHECK is set

Mamzi Bayatpour  mbayatpour@nvidia.com () · janjust · Mamzi Bayatpour  mbayatpour@nvidia.com () · commit f7f928deaf7d · 2022-06-22T13:58:47.000-07:00
Signed-off-by: Mamzi Bayatpour  &lt;mbayatpour@nvidia.com&gt;
Co-authored-by: Tomislav Janjusic &lt;tomislavj@nvidia.com&gt;
diff --git a/ompi/mca/osc/ucx/osc_ucx_passive_target.c b/ompi/mca/osc/ucx/osc_ucx_passive_target.c
@@ -124,6 +124,11 @@ int ompi_osc_ucx_lock(int lock_type, int target, int mpi_assert, struct ompi_win
         }
     } else {
         lock->is_nocheck = true;
+        if (lock_type == MPI_LOCK_EXCLUSIVE) {
+            lock->type = LOCK_EXCLUSIVE;
+        } else {
+            lock->type = LOCK_SHARED;
+        }
     }
 
     if (ret == OMPI_SUCCESS) {

Original file line number	Diff line number	Diff line change
`@@ -124,6 +124,11 @@ int ompi_osc_ucx_lock(int lock_type, int target, int mpi_assert, struct ompi_win`
`124`	`124`	`}`
`125`	`125`	`} else {`
`126`	`126`	`lock->is_nocheck = true;`
	`127`	`+ if (lock_type == MPI_LOCK_EXCLUSIVE) {`
	`128`	`+ lock->type = LOCK_EXCLUSIVE;`
	`129`	`+ } else {`
	`130`	`+ lock->type = LOCK_SHARED;`
	`131`	`+ }`
`127`	`132`	`}`
`128`	`133`
`129`	`134`	`if (ret == OMPI_SUCCESS) {`