def METHOD_NAME(test_case):
    _test_rmsnorm(
        test_case,
        shape=[16, 1024 * 1024],
        normalized_shape=[1024 * 1024],
        dtype=flow.float16,
    )
    _test_rmsnorm(
        test_case, shape=[8, 1024], normalized_shape=[1024], dtype=flow.double
    )