def METHOD_NAME():
    seed = 123
    np.random.seed(seed)
    torch.manual_seed(seed)
    random.seed(seed)
    min_run_time = 10
    batch_size = 64
    num_heads = 32
    max_seq_len = 256
    embed_dim = 1024
    dtype = torch.bfloat16
    print(
        f"Running timing for batch_size={batch_size} max_sequence_len={max_seq_len} "
        f"num_heads={num_heads} embed_dimension={embed_dim} dtype={dtype}"
    )
    run_timing(min_run_time, batch_size, embed_dim, num_heads, max_seq_len, dtype)