added maximum launch size checking on pipeline launch

Author: Felix Igelbrink

optix/context.pyx

@@ -21,6 +21,9 @@ cdef void context_log_cb(unsigned int level, const char * tag, const char * mess
         cb_object(level, tag.decode(), message.decode())
 
 
+cdef size_t MAXIMUM_LAUNCH_SIZE = 2**30
+
+
 cdef class DeviceContext(OptixObject):
     """
     Represents an OptiX context on a single device. This class wraps the OptixDeviceContext struct.
@@ -233,6 +236,14 @@ cdef class DeviceContext(OptixObject):
         """
         return self._device
 
+    @property
+    def maximum_launch_size(self):
+        """
+        The maximum launch size (over width, height, depth) of a single pipeline launch
+        """
+        return MAXIMUM_LAUNCH_SIZE
+
+
     def _repr_details(self):
         return f"device id {self._device.id}"
 

optix/pipeline.pyx

@@ -405,6 +405,10 @@ cdef class Pipeline(OptixContextObject):
         for i in range(len(dimensions)):
             c_dims[i] = dimensions[i]
 
+        cdef size_t total_launch_size = c_dims[0] * c_dims[1] * c_dims[2]
+        if total_launch_size > self.context.maximum_launch_size:
+            raise ValueError(f"Requested launch size of {total_launch_size} is larger than the limit of {self.context.maximum_launch_size} set by OptiX.")
+
         cdef size_t c_stream = 0
         if stream is not None:
             c_stream = stream.ptr