spirv: Use nir_builder for control flow