Fix PyOpenCL local memory.

diku-dk · Aug 4, 2023 · a721bdf · a721bdf
1 parent 0539183
commit a721bdf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/Futhark/CodeGen/Backends/PyOpenCL.hs b/src/Futhark/CodeGen/Backends/PyOpenCL.hs
@@ -261,7 +261,9 @@ launchKernel kernel_name safety kernel_dims workgroup_dims local_memory args = d
           ]
   Py.stm . Exp $
     Py.simpleCall (T.unpack $ kernel_name' <> ".set_args") $
-      [local_memory] ++ failure_args ++ args'
+      [Py.simpleCall "cl.LocalMemory" [Py.simpleCall "max" [local_memory, Integer 1]]]
+        ++ failure_args
+        ++ args'
   Py.stm . Exp $
     Py.simpleCall
       "cl.enqueue_nd_range_kernel"