Added wait_ip_interrupt(timeout) functionality on edge side (#8611)

Signed-off-by: Manoj Takasi <[email protected]>
Co-authored-by: Manoj Takasi <[email protected]>

src/runtime_src/core/edge/user/device_linux.cpp

@@ -17,6 +17,7 @@
 #include "core/edge/user/aie/profile_object.h"
 #include <map>
 #include <memory>
+#include <poll.h>
 #include <string>
 
 #include <fcntl.h>
@@ -1293,4 +1294,26 @@ wait_ip_interrupt(xclInterruptNotifyHandle handle)
     throw error(errno, "wait_ip_interrupt failed POSIX read");
 }
 
+std::cv_status
+device_linux::
+wait_ip_interrupt(xclInterruptNotifyHandle handle, int32_t timeout)
+{
+  struct pollfd pfd = {.fd=handle, .events=POLLIN};
+  int32_t ret = 0;
+
+  //Checking for only one fd; Only of one CU
+  //Timeout value in milli seconds
+  ret = ::poll(&pfd, 1, timeout);
+  if (ret < 0)
+    throw error(errno, "wait_timeout: failed POSIX poll");
+
+  if (ret == 0) //Timeout occured
+    return std::cv_status::timeout;
+
+  if (pfd.revents & POLLIN) //Interrupt received
+    return std::cv_status::no_timeout;
+
+  throw error(-EINVAL, boost::str(boost::format("wait_timeout: POSIX poll unexpected event: %d")  % pfd.revents));
+}
+
 } // xrt_core

src/runtime_src/core/edge/user/device_linux.h

@@ -91,6 +91,9 @@ class device_linux : public shim<device_edge>
   virtual void
   wait_ip_interrupt(xclInterruptNotifyHandle);
 
+  virtual std::cv_status
+  wait_ip_interrupt(xclInterruptNotifyHandle, int32_t timeout);
+
   virtual std::unique_ptr<hwctx_handle>
   create_hw_context(const xrt::uuid& xclbin_uuid,
                     const xrt::hw_context::cfg_param_type& cfg_param,