Some bug fixes on compute shaders.