ac/nir, radv, radeonsi: Switch to using ac_shader_args