hero: Added printf during offload

pulp-platform · May 13, 2024 · a55c3e2 · a55c3e2
1 parent df91a5e
commit a55c3e2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/openmp/libomptarget/plugins/hero/rtl.cpp b/openmp/libomptarget/plugins/hero/rtl.cpp
@@ -486,7 +486,14 @@ int32_t __tgt_rtl_run_target_team_region(int32_t device_id, void *tgt_entry_ptr,
   hero_dev_mbox_write(hd, num_miss_handler_threads);
 
   uint32_t ret[2];
-  while (hero_dev_mbox_read(hd, (unsigned int *)&ret[0], 1));
+  ret[0] = 0;
+  while(ret[0] != MBOX_DEVICE_DONE) {
+    while (hero_dev_mbox_read(hd, (unsigned int *)&ret[0], 1));
+    if(ret[0] == MBOX_DEVICE_PRINT) {
+      while (hero_dev_mbox_read(hd, (unsigned int *)&ret[0], 1));
+      printf("%c", ret[0]);
+    }
+  }
   assert(ret[0] == MBOX_DEVICE_DONE &&
          "Software mailbox protocol failure: Expected MBOX_DEVICE_DONE.");
   while (hero_dev_mbox_read(hd, (unsigned int *)&ret[1], 1));