def METHOD_NAME(test_case):
    V = 10
    H = 4
    S = 6
    P = flow.placement("cuda", [[0, 1], [2, 3]])
    wte = flow.nn.Parameter(
        flow.empty(
            (V, H),
            dtype=flow.float32,
            placement=P,
            sbp=[flow.sbp.broadcast, flow.sbp.split(0)],
        )
    )
    wpe = flow.nn.Parameter(
        flow.empty(
            (S, H),
            dtype=flow.float32,
            placement=P,
            sbp=[flow.sbp.broadcast, flow.sbp.broadcast],
        )
    )
    flow.nn.init.normal_(wte, std=0.02)
    flow.nn.init.normal_(wpe, std=0.02)