nir: nir_builder_init_simple_shader() ergonomics

Having had to write another call to nir_builder_init_simple_shader(), I was bothered by some bits of shader setup that could be simplified. Here's an attempt to make NIR shader writing in C just a little nicer.