[Bugfix][CI] Fix wrong residual shape in TestFusedAddRMSNorm.example_inputs...

[Bugfix][CI] Fix wrong residual shape in TestFusedAddRMSNorm.example_inputs that causes flaky test (#40629)
Signed-off-by: Zhang Jian <jianmusings@gmail.com>

tests/compile/passes/test_functionalization.py

@@ -117,9 +117,9 @@ class TestFusedAddRMSNorm(torch.nn.Module):
         else:
             return norm_output, residual_output
 
-    def example_inputs(self, batch_size=8, hidden_size=16, seq_len=16):
-        hidden_states = torch.randn((batch_size * seq_len, hidden_size))
-        residual = torch.randn((batch_size * seq_len, hidden_size))
+    def example_inputs(self, batch_size=8, seq_len=16):
+        hidden_states = torch.randn((batch_size * seq_len, self.hidden_size))
+        residual = torch.randn((batch_size * seq_len, self.intermediate_size))
         return (hidden_states, residual)
 
     def ops_in_model(self, do_fusion):