Implement timeout in util_wait_yield_run()

Signed-off-by: Franz Pöschel <[email protected]>
ofiwg · Dec 15, 2023 · c545402 · c545402
1 parent b396355
commit c545402
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/prov/util/src/util_wait.c b/prov/util/src/util_wait.c
@@ -593,10 +593,15 @@ static int util_wait_yield_run(struct fid_wait *wait_fid, int timeout)
 {
 	struct util_wait_yield *wait;
 	struct ofi_wait_fid_entry *fid_entry;
+	uint64_t endtime;
 	int ret = 0;
 
 	wait = container_of(wait_fid, struct util_wait_yield, util_wait.wait_fid);
+	endtime = ofi_timeout_time(timeout);
+
 	while (!wait->signal) {
+		if (ofi_adjust_timeout(endtime, &timeout))
+			return -FI_ETIMEDOUT;
 		ofi_mutex_lock(&wait->util_wait.lock);
 		dlist_foreach_container(&wait->util_wait.fid_list,
 					struct ofi_wait_fid_entry,