[Fix] TE RMSNorm Triton Kernel Optimization #778

Job	Run time
determine_level	3s
CI Level 3 / Select Docker Image	5s
CI Level 3 / build / Build ROCm Docker image and TransformerEngine wheels	24m 44s
CI Level 3 / sGPU Tests (mi35x)	1h 24m 5s
CI Level 3 / sGPU Tests (mi30x)	1h 23m 52s
CI Level 3 / mGPU Torch (mi35x)	34m 32s
CI Level 3 / mGPU JAX (mi30x)	55m 57s
CI Level 3 / mGPU JAX (mi35x)	37m 8s
CI Level 3 / mGPU Torch (mi30x)	59m 36s
	6h 20m 2s

Provide feedback