def METHOD_NAME(test_case):
    args, unknown_args = _parse_args()
    placement = flow.placement.all("cuda")
    sbp = [flow.sbp.broadcast,] * len(placement.ranks.shape)
    _test_alexnet_graph(test_case, args, placement, sbp)