UCT/DEVICE: fix lock

JeynmannZ · JeynmannZ · commit 2fe74a417b01 · 2025-11-10T12:38:22.000+02:00
diff --git a/src/uct/ib/mlx5/gdaki/gdaki.cuh b/src/uct/ib/mlx5/gdaki/gdaki.cuh
@@ -199,10 +199,12 @@ UCS_F_DEVICE void uct_rc_mlx5_gda_db(uct_rc_gdaki_dev_ep_t *ep,
     }
 
     if (no_delay) {
+        uct_rc_mlx5_gda_lock(&ep->sq_lock);
         const uint64_t ready_index = ep->sq_ready_index;
         uct_rc_mlx5_gda_ring_db(ep, ready_index);
         uct_rc_mlx5_gda_update_dbr(ep, ready_index);
         uct_rc_mlx5_gda_ring_db(ep, ready_index);
+        uct_rc_mlx5_gda_unlock(&ep->sq_lock);
         return;
     }
 

Original file line number	Diff line number	Diff line change
`@@ -199,10 +199,12 @@ UCS_F_DEVICE void uct_rc_mlx5_gda_db(uct_rc_gdaki_dev_ep_t *ep,`
`199`	`199`	`}`
`200`	`200`
`201`	`201`	`if (no_delay) {`
	`202`	`+ uct_rc_mlx5_gda_lock(&ep->sq_lock);`
`202`	`203`	`const uint64_t ready_index = ep->sq_ready_index;`
`203`	`204`	`uct_rc_mlx5_gda_ring_db(ep, ready_index);`
`204`	`205`	`uct_rc_mlx5_gda_update_dbr(ep, ready_index);`
`205`	`206`	`uct_rc_mlx5_gda_ring_db(ep, ready_index);`
	`207`	`+ uct_rc_mlx5_gda_unlock(&ep->sq_lock);`
`206`	`208`	`return;`
`207`	`209`	`}`
`208`	`210`