nir/algebraic: Refactor codegen a bit