Handle functions without outputs in JITLinker

pytensor/link/basic.py

@@ -656,6 +656,8 @@ class JITLinker(PerformLinker):
         thunk_outputs = [storage_map[n] for n in self.fgraph.outputs]
         fgraph_jit = self.jit_compile(converted_fgraph)
 
+        if thunk_outputs:
+
             def thunk(
                 fgraph_jit=fgraph_jit,
                 thunk_inputs=thunk_inputs,
@@ -672,6 +674,20 @@ class JITLinker(PerformLinker):
                 for o_storage, o_val in zip(thunk_outputs, outputs):
                     o_storage[0] = o_val
 
+        else:
+            # Edge case - functions without outputs
+            def thunk(
+                fgraph_jit=fgraph_jit,
+                thunk_inputs=thunk_inputs,
+                thunk_outputs=thunk_outputs,
+            ):
+                try:
+                    res = fgraph_jit(*(x[0] for x in thunk_inputs))
+                except Exception:
+                    raise_with_op(self.fgraph, output_nodes[0], thunk)
+                assert res is None
+                return thunk_outputs
+
         thunk.inputs = thunk_inputs
         thunk.outputs = thunk_outputs
         thunk.lazy = False