train/compute/pt/pytorch_cutlass.py [41:52]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    warmups = args.warmups
    steps = args.steps

    dt = torch.float32
    if (dtype == "float16" or dtype == "half"):
        dt = torch.float16
    elif (dtype == "bfloat16"):
        dt = torch.bfloat16

    torch.manual_seed(0)

    elap = 0.0
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



train/compute/pt/pytorch_gemm.py [60:71]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    warmups = args.warmups
    steps = args.steps

    dt = torch.float32
    if (dtype == "float16" or dtype == "half"):
        dt = torch.float16
    elif (dtype == "bfloat16"):
        dt = torch.bfloat16

    torch.manual_seed(0)

    elap = 0.0
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -