shuffle.py

sizes = [
    128,
    512,
    2 * 1024,
    8 * 1024,
    32 * 1024,
    128 * 1024,
    512 * 1024,
    2 * 1024 * 1024,
    8 * 1024 * 1024,
    32 * 1024 * 1024,
    128 * 1024 * 1024,
]

for n in sizes:
    random_state = cupy.random.RandomState(seed=1)
    array = cupy.arange(n)
    name = f'size = {n}'
    perf = cupyx.time.repeat(random_state.shuffle, (array,), name=name, max_duration=5)
    print(perf)