kernelforce reprogram if kernel device has device global with init_mode reprogram

Author: sophimao

src/acl_kernel.cpp

@@ -3052,23 +3052,34 @@ int acl_submit_kernel_device_op(cl_event event) {
   // to free up old operation slots.
   acl_forget_proposed_device_ops(doq);
 
-  bool need_reprogram = true;
-  if (device->last_bin) {
-    // compare hash of last program that went through device op queue and the
-    // program required by kernel
-    need_reprogram =
-        device->last_bin->get_devdef().autodiscovery_def.binary_rand_hash !=
-        dev_bin->get_devdef().autodiscovery_def.binary_rand_hash;
-  } else {
-    // compare hash of program that is on the device and the program required by
-    // kernel
-    need_reprogram = device->def.autodiscovery_def.binary_rand_hash !=
-                     dev_bin->get_devdef().autodiscovery_def.binary_rand_hash;
+  // Check if any of the device globals have init_mode:reprogram
+  bool has_reprogram_dev_global = false;
+  std::unordered_map<std::string, acl_device_global_mem_def_t>
+      device_global_mem_defs =
+          device->def.autodiscovery_def.device_global_mem_defs;
+  for (auto &it : device_global_mem_defs) {
+    if (it.second.init_mode == ACL_DEVICE_GLOBAL_INIT_MODE_REPROGRAM) {
+      has_reprogram_dev_global = true;
+      break;
+    }
   }
 
-  if (event->context->split_kernel) {
-    // Always reprogram in split kernel mode. This is a temporary workaround.
-    need_reprogram = true;
+  bool need_reprogram = true;
+  // Force reprogram if there is device global with init_mode:reprogram
+  // or if this is the split kernel mode (split kernel workaround)
+  if (!has_reprogram_dev_global && !event->context->split_kernel) {
+    if (device->last_bin) {
+      // compare hash of last program that went through device op queue and the
+      // program required by kernel
+      need_reprogram =
+          device->last_bin->get_devdef().autodiscovery_def.binary_rand_hash !=
+          dev_bin->get_devdef().autodiscovery_def.binary_rand_hash;
+    } else {
+      // compare hash of program that is on the device and the program required
+      // by kernel
+      need_reprogram = device->def.autodiscovery_def.binary_rand_hash !=
+                       dev_bin->get_devdef().autodiscovery_def.binary_rand_hash;
+    }
   }
 
   if (need_reprogram) {