1db4255e8b19a322434fa2fce398b73b5757f764,examples/1d/compute_speed.py,,,#,19

Before Change

    torch.cuda.synchronize()

    t_start = time.time()
    for _ in range(times):
        scattering(x_data)

    torch.cuda.synchronize()

    t_elapsed = time.time() - t_start

    fmt_str = "Elapsed time: {:2f} [s / {:d} evals], avg: {:.2f} (s/batch)"
    print(fmt_str.format(t_elapsed, times, t_elapsed/times))

After Change

if backend.NAME == "torch":
    devices = ["cpu", "gpu"]
elif backend.NAME == "skcuda":
    devices = ["gpu"]

//////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

Instances