def METHOD_NAME(test_case):
    os.environ["ONEFLOW_BOXING_DISABLE_MIDDLE_NODE_AND_CHECK"] = "0"
    os.environ["ONEFLOW_BOXING_ENABLE_GENERAL_BASIC_COMMUNICATION"] = "0"
    x = flow.ones(
        4,
        12,
        4,
        sbp=[flow.sbp.broadcast, flow.sbp.broadcast],
        placement=flow.placement(
            type="cuda", ranks=np.array(range(4)).reshape(2, 2)
        ),
    )
    model_diff_hierarchy = _TestModuleDiffHierarchy()
    graph_diff_hierarchy = _TestGraph(model_diff_hierarchy)
    y = graph_diff_hierarchy(x)