def _dispatch_compute(self, task):
        """
        Call me per frame when you want to access the depth texture result with cuda enabled
        """
        work_group_x = int(np.ceil(self.depth_tex.getXSize() / self.shader_local_size[0]))
        work_group_y = int(np.ceil(self.depth_tex.getYSize() / self.shader_local_size[1]))
        self.engine.graphicsEngine.dispatch_compute(
            (work_group_x, work_group_y, 1), self.compute_node.get_attrib(ShaderAttrib), self.engine.win.get_gsg()
        )
        # self.engine.graphicsEngine.extractTextureData(self.output_tex, self.engine.win.get_gsg())
        # self.output_tex.write("{}.png".format(self.engine.episode_step))
        return task.cont

        Call me per frame when you want to access the depth texture result with cuda enabled
        
metadrive/component/sensors/depth_camera.py