nir/lower_amul: Add a variable mode check