bench_rmsnorm.py 12 KB