Revert "Improve RMSNorm to support 2D inputs" #1029

Job	Run time
Dependency Review	4s
CodeQL Scan	3m 8s
Calculate Tests To Trigger	5s
Test / gsm8k_small_g3_tp1	8m 47s
Test / gsm8k_small_g3_tp2	9m 2s
Test / gsm8k_small_g2_tp1	12m 40s
Test / gsm8k_small_g2_tp2	11m 20s
Test / gsm8k_large_g3_tp2	6m 18s
Test / gsm8k_large_g2_tp4	10m 20s
Test / gsm8k_small_g3_tp1_fp8	5m 21s
Test / gsm8k_small_g3_tp2_fp8	6m 37s
Test / gsm8k_small_g3_tp1_mss	7m 26s
Test / gsm8k_small_g2_tp1_mss	10m 5s
Test / gsm8k_small_g3_tp2_mss	7m 5s
Test / gsm8k_small_g2_tp2_mss	8m 29s
Test / gsm8k_small_g2_tp1_spec_decode	11m 19s
Test / gsm8k_small_g2_tp1_mlp_spec_decode	8m 15s
Test / gsm8k_small_g2_tp1_medusa_spec_decode	9m 40s
Test / gsm8k_small_g2_tp1_eagle_spec_decode	9m 42s
Test / test_llama_lora	7m 5s
Test / test_multilora	6m 49s
Test / test_long_context	6m 8s
Test / multimodal_small_g3_tp1	7m 20s
Test / multimodal_small_g3_tp2	6m 30s
Test / multimodal_small_g3_tp1_mss	7m 40s
Test / multimodal_small_g3_tp2_mss	6m 43s
Test / test_awq	8m 26s
Test / test_gptq	9m 0s
Test / test_lazy_outlines	5m 59s
Test / test_guided_generate	11m 44s
	3h 49m 7s

Provide feedback