UCT/DEVICE: fix lock

JeynmannZ · JeynmannZ · commit 01a58e0a1d5e · 2025-11-05T20:19:48.000+02:00
diff --git a/src/uct/ib/mlx5/gdaki/gdaki.cuh b/src/uct/ib/mlx5/gdaki/gdaki.cuh
@@ -274,10 +274,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
`@@ -274,10 +274,12 @@ UCS_F_DEVICE void uct_rc_mlx5_gda_db(uct_rc_gdaki_dev_ep_t *ep,`
`274`	`274`	`}`
`275`	`275`
`276`	`276`	`if (no_delay) {`
	`277`	`+ uct_rc_mlx5_gda_lock(&ep->sq_lock);`
`277`	`278`	`const uint64_t ready_index = ep->sq_ready_index;`
`278`	`279`	`uct_rc_mlx5_gda_ring_db(ep, ready_index);`
`279`	`280`	`uct_rc_mlx5_gda_update_dbr(ep, ready_index);`
`280`	`281`	`uct_rc_mlx5_gda_ring_db(ep, ready_index);`
	`282`	`+ uct_rc_mlx5_gda_unlock(&ep->sq_lock);`
`281`	`283`	`return;`
`282`	`284`	`}`
`283`	`285`