Store hasSpecialCaptures in kernel information

We are currently only able to skip argument extraction in the case
where a lambda has no special captures. We can only detect this
while we have the kernel type name, and must carry it through until
we call handler::finalize().

Signed-off-by: John Pennycook <[email protected]>

Author: Pennycook

sycl/include/sycl/handler.hpp

@@ -799,7 +799,8 @@ class __SYCL_EXPORT handler {
       setKernelInfo((void *)MHostKernel->getPtr(),
                     detail::getKernelNumParams<KernelName>(),
                     &(detail::getKernelParamDesc<KernelName>),
-                    detail::isKernelESIMD<KernelName>());
+                    detail::isKernelESIMD<KernelName>(),
+                    detail::hasSpecialCaptures<KernelName>());
 
       MKernelName = detail::getKernelName<KernelName>();
     } else {
@@ -3763,7 +3764,7 @@ class __SYCL_EXPORT handler {
 
   void setKernelInfo(void *KernelFuncPtr, int KernelNumArgs,
                      detail::kernel_param_desc_t (*KernelParamDescGetter)(int),
-                     bool KernelIsESIMD);
+                     bool KernelIsESIMD, bool KernelHasSpecialCaptures);
 
   friend class detail::HandlerAccess;
 

sycl/source/detail/handler_impl.hpp

@@ -204,6 +204,7 @@ class handler_impl {
   int MKernelNumArgs = 0;
   detail::kernel_param_desc_t (*MKernelParamDescGetter)(int) = nullptr;
   bool MKernelIsESIMD = false;
+  bool MKernelHasSpecialCaptures = false;
 };
 
 } // namespace detail

sycl/source/handler.cpp

@@ -2266,11 +2266,12 @@ void handler::setNDRangeDescriptorPadded(sycl::range<3> NumWorkItems,
 void handler::setKernelInfo(
     void *KernelFuncPtr, int KernelNumArgs,
     detail::kernel_param_desc_t (*KernelParamDescGetter)(int),
-    bool KernelIsESIMD) {
+    bool KernelIsESIMD, bool KernelHasSpecialCaptures) {
   impl->MKernelFuncPtr = KernelFuncPtr;
   impl->MKernelNumArgs = KernelNumArgs;
   impl->MKernelParamDescGetter = KernelParamDescGetter;
   impl->MKernelIsESIMD = KernelIsESIMD;
+  impl->MKernelHasSpecialCaptures = KernelHasSpecialCaptures;
 }
 
 void handler::saveCodeLoc(detail::code_location CodeLoc, bool IsTopCodeLoc) {