Performance degradation by 4 times for dpnp.dot function for current case. Affected pairwise_distance workload dpnp version #1669 ``` python In [1]: import dpnp In [2]: a = dpnp.ones((21846, 3), device='cpu') In [3]: b = dpnp.ones((21846, 3), device='cpu') In [4]: c = dpnp.ones((21846, 21846), device='cpu') In [5]: %timeit dpnp.dot(a, b.T, c) 1.18 s ± 59.2 ms per loop (mean ± std. dev. of 7 runs, 1 loop each) ``` ``` python In [1]: import numpy In [2]: a_np = numpy.ones((21846, 3)) In [3]: b_np = numpy.ones((21846, 3)) In [4]: c_np = numpy.ones((21846, 21846)) In [5]: %timeit numpy.dot(a_np, b_np.T, c_np) 457 ms ± 9.27 ms per loop (mean ± std. dev. of 7 runs, 1 loop each) ``` dpnp version 0.13.1dev3+15 ``` python In [1]: import dpnp In [2]: a = dpnp.ones((21846, 3), device='cpu') In [3]: b = dpnp.ones((21846, 3), device='cpu') In [4]: c = dpnp.ones((21846, 21846), device='cpu') In [5]: %timeit dpnp.dot(a, b.T, c) 288 ms ± 12.1 ms per loop (mean ± std. dev. of 7 runs, 1 loop each) ```
Activity
vtavana commentedon Feb 7, 2024
resolved in #1694