IntelPython / dpnp

Data Parallel Extension for NumPy
BSD 2-Clause "Simplified" License
94 stars 21 forks source link

Implement gesv_batch via gesv call #1877

Open vlad-perevezentsev opened 1 month ago

vlad-perevezentsev commented 1 month ago

This PR suggests adding gesv_batch implementation as a separate function and provide iteration of the batch array within it to avoid allocation of each 2D array before calling gesv to improve performance and avoid serialization of all host tasks in the queue in MKL call;

Performance of dpnp.solve on Iris Xe image

github-actions[bot] commented 1 month ago

View rendered docs @ https://intelpython.github.io/dpnp/pull/1877/index.html