mesa/cs: Add a MESA_SHADER_COMPUTE stage and update switch statements.