Skip to content

dpnp.dot performance degradation by 4 times #1693

Description

@npolina4

Performance degradation by 4 times for dpnp.dot function for current case.
Affected pairwise_distance workload

dpnp version #1669

In [1]: importdpnpIn [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]: %timeitdpnp.dot(a, b.T, c)
1.18s ± 59.2msperloop (mean ± std. dev. of7runs, 1loopeach)
In [1]: importnumpyIn [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]: %timeitnumpy.dot(a_np, b_np.T, c_np)
457ms ± 9.27msperloop (mean ± std. dev. of7runs, 1loopeach)

dpnp version 0.13.1dev3+15

In [1]: importdpnpIn [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]: %timeitdpnp.dot(a, b.T, c)
288ms ± 12.1msperloop (mean ± std. dev. of7runs, 1loopeach)

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions