def METHOD_NAME(test_case):
    arg_dict = OrderedDict()
    arg_dict["device"] = ["cuda"]
    arg_dict["x_shape"] = [(4, 4)]
    arg_dict["tensor_num"] = [4, 6]
    arg_dict["learning_rate"] = [1, 1e-3]
    arg_dict["train_iters"] = [10]
    arg_dict["weight_decay"] = [0.0, 1e-3]
    for arg in GenArgList(arg_dict):
        compare_with_numpy_sgd(test_case, *arg)