refactor: RMSNorm

[zhangzefeng92]

refactor rms norm op, and rotary_embeding and mha.

[lljbash]

有几点关于 python module 的东西：

1. 如果文件夹当 module，那么必须要有 __init__.py
2. 如果不是多个文件，那么不需要文件夹，直接用文件当 module
3. 无论什么情况，被 import 的入口都要有 __all__

此外，长期规划应该是 cpp 对接 diopi，python 层对接 python 层，那么我建议把 common 改叫 ops

[lljbash]

Suggested change

	all = ["rms_norm_out", "rms_norm", "rms_norm_backward_out", "rms_norm_backward"]
	__all__ = ["rms_norm_out", "rms_norm", "rms_norm_backward_out", "rms_norm_backward"]

[lljbash]

这个签名需要修改

那个头文件可以完全删除

[lljbash]

Suggested change

	__all__ = ["mha", "rotary", RMSNorm, RMSNormWithNormalizedShape]
	__all__ = ["mha", "rotary", "RMSNorm", "RMSNormWithNormalizedShape"]

[lljbash]

common 叫 ops 比较好吧

[yangbofun]

common 叫 ops 比较好吧

这个等后续看看，感觉common意思还更直观。