Moved check for relaxed allocation limit extension to DeviceGPU constructor.

Author: stefanatwork

kernels/common/device.cpp

@@ -665,8 +665,12 @@ namespace embree
     bool ze_rtas_builder = false;
     for (uint32_t i=0; i<extensions.size(); i++)
     {
-      if (strncmp("ZE_extension_rtas",extensions[i].name,sizeof(extensions[i].name)) == 0)
+      if (strncmp("ZE_extension_rtas",extensions[i].name,sizeof(extensions[i].name)) == 0) {
         ze_rtas_builder = true;
+      }
+      if (strncmp("ZE_experimental_relaxed_allocation_limits", extensions[i].name, sizeof(extensions[i].name)) == 0) {
+        hasRelaxedAllocationLimits = true;
+      }
     }
     if (!ze_rtas_builder)
       throw_RTCError(RTC_ERROR_LEVEL_ZERO_RAYTRACING_SUPPORT_MISSING, "ZE_extension_rtas extension not found. Please install a recent driver. On Linux, make sure that the package intel-level-zero-gpu-raytracing is installed");

kernels/common/device.h

@@ -209,7 +209,8 @@ namespace embree
 
     unsigned int gpu_maxWorkGroupSize;
     unsigned int gpu_maxComputeUnits;
-
+    bool hasRelaxedAllocationLimits = false;
+ 
   public:
     void* dispatchGlobalsPtr = nullptr;
 
@@ -218,6 +219,8 @@ namespace embree
     inline sycl::context &getGPUContext() { return gpu_context; }    
 
     inline unsigned int getGPUMaxWorkGroupSize() { return gpu_maxWorkGroupSize; }
+    inline bool relaxedAllocationLimitsSupported() const { return hasRelaxedAllocationLimits; }
+
 
     void init_rthw_level_zero();
     void init_rthw_opencl();

kernels/sycl/rthwif_embree_builder.cpp

@@ -214,8 +214,13 @@ namespace embree
     relaxed.stype = ZE_STRUCTURE_TYPE_RELAXED_ALLOCATION_LIMITS_EXP_DESC;
     relaxed.pNext = &rt_desc;
     relaxed.flags = ZE_RELAXED_ALLOCATION_LIMITS_EXP_FLAG_MAX_SIZE;
-
-    const bool hasRelaxedAllocationLimits = hasDriverExtension(hDriver, ZE_RELAXED_ALLOCATION_LIMITS_EXP_NAME);
+    
+    bool hasRelaxedAllocationLimits = false;
+    if (embree_device) {
+      DeviceGPU* gpu_device = dynamic_cast<DeviceGPU*>(embree_device);
+      if (gpu_device)
+        hasRelaxedAllocationLimits = gpu_device->relaxedAllocationLimitsSupported();
+    }
 
     ze_memory_compression_hints_ext_desc_t compressed;
     compressed.stype = ZE_STRUCTURE_TYPE_MEMORY_COMPRESSION_HINTS_EXT_DESC;