Fix urEnqueueKernelLaunchWithArgsExp() in asan_ddi.cpp

Author: ldorau

unified-runtime/source/loader/layers/sanitizer/asan/asan_ddi.cpp

@@ -1749,10 +1749,26 @@ __urdlllocal ur_result_t UR_APICALL urEnqueueKernelLaunchWithArgsExp(
 
   UR_CALL(getAsanInterceptor()->preLaunchKernel(hKernel, hQueue, LaunchInfo));
 
-  UR_CALL(getContext()->urDdiTable.EnqueueExp.pfnKernelLaunchWithArgsExp(
+  /*
+    UR_CALL(getContext()->urDdiTable.EnqueueExp.pfnKernelLaunchWithArgsExp(
+        hQueue, hKernel, workDim, pGlobalWorkOffset, pGlobalWorkSize,
+        LaunchInfo.LocalWorkSize.data(), 0, nullptr, numPropsInLaunchPropList,
+        launchPropList, numEventsInWaitList, phEventWaitList, phEvent));
+  */
+
+  ur_result_t UrRes = getContext()->urDdiTable.Enqueue.pfnKernelLaunch(
       hQueue, hKernel, workDim, pGlobalWorkOffset, pGlobalWorkSize,
-      LaunchInfo.LocalWorkSize.data(), 0, nullptr, numPropsInLaunchPropList,
-      launchPropList, numEventsInWaitList, phEventWaitList, phEvent));
+      LaunchInfo.LocalWorkSize.data(), numPropsInLaunchPropList, launchPropList,
+      numEventsInWaitList, phEventWaitList, phEvent);
+  if (UrRes != UR_RESULT_SUCCESS) {
+    if (UrRes == UR_RESULT_ERROR_OUT_OF_DEVICE_MEMORY) {
+      UR_LOG_L(
+          getContext()->logger, ERR,
+          "urEnqueueKernelLaunch failed due to out of device memory, maybe "
+          "SLM is fully used.");
+    }
+    return UrRes;
+  }
 
   UR_CALL(getAsanInterceptor()->postLaunchKernel(hKernel, hQueue, LaunchInfo));