Correctly set the work group size

yt-project · Dec 6, 2024 · 19f82f5 · 19f82f5
1 parent a472dff
commit 19f82f5
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/yt_idv/scene_annotations/block_histogram.py b/yt_idv/scene_annotations/block_histogram.py
@@ -51,11 +51,12 @@ def compute(self, scene, program):
                         # architecture, I guess.  That aspect of running compute
                         # shaders, CUDA, etc, is one of my absolute least favorite
                         # parts.
-                        GL.glDispatchCompute(self.bins, 1, 1)
-        # arr = self.output_data.read_as_image()
-        # print(
-        # f"histogram: {arr=} => {arr.sum()=} with {total=} for a ratio of {arr.sum()/total}"
-        # )
+                        GL.glDispatchCompute(
+                            bitmap_tex.data.shape[0],
+                            bitmap_tex.data.shape[1],
+                            bitmap_tex.data.shape[2],
+                        )
+                        GL.glMemoryBarrier(GL.GL_SHADER_IMAGE_ACCESS_BARRIER_BIT)
 
     def draw(self, scene, program):
         # This will probably need to have somewhere to draw the darn thing.  So