-
Notifications
You must be signed in to change notification settings - Fork 22
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Improve performance of dpnp.matmul
and dpnp.dot
with out
keyword
#1694
Conversation
View rendered docs @ https://intelpython.github.io/dpnp/pull//index.html |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Performance improved.
Thanks @vtavana!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you @vtavana for resolving the issue so quickly
In this PR, implementation of
dpnp.matmul
anddpnp.dot
is updated to improve the performance whenout
keyword is present and its feature matches the appropriate shape, dtype, sycl_queue, and usm_type that is needed for performing the calculation in the corresponding OneMKL BLAS routines.Sample timing including this update
Sample timing excluding this update