Quick benchmark for cupy.percentile

percentile_benchmark.py

import cupy
from cupyx.time import repeat



cupy.random.seed(0)
for size in (5, 50, 500, 5000, 50000, int(5e6)):
    a = cupy.random.rand(size)
    q = cupy.linspace(0, 100, 3)
    def f():
        cupy.percentile(a, q, interpolation='linear')

    print(f"For array of length {size}")
    print(repeat(f))
    print()