I changeed self.norm in mamba2_simple from RMS to Layernorm but went wrong

[LuoMayao]

I changeed self.norm in mamba2_simple from RMS to Layernorm but went wrong.I was told the fllowing error:
subprocess.CalledProcessError: Command '['/usr/bin/gcc', '/tmp/tmpeniti2ke/main.c', '-O3', '-I/data3/ipad_3d/miniconda3/envs/pointcept_ex/lib/python3.10/site-packages/triton/common/../third_party/cuda/include', '-I/data3/ipad_3d/miniconda3/envs/pointcept_ex/include/python3.10', '-I/tmp/tmpeniti2ke', '-shared', '-fPIC', '-lcuda', '-o', '/tmp/tmpeniti2ke/_layer_norm_fwd_1pass_kernel.cpython-310-x86_64-linux-gnu.so', '-L/lib/x86_64-linux-gnu', '-L/lib/i386-linux-gnu', '-L/lib/i386-linux-gnu']' returned non-zero exit status 1.

I didn't open use_mem_eff_path, but I went successfully when I use RMSNorm.