Add some memory barrier when using compute shaders