-
Notifications
You must be signed in to change notification settings - Fork 23
Update dpnp.isclose
with scalar-specific SYCL kernels
#2540
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
Conversation
View rendered docs @ https://intelpython.github.io/dpnp/index.html |
Array API standard conformance tests for dpnp=0.19.0dev3=py313h509198e_61 ran successfully. |
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
dpnp/backend/extensions/ufunc/elementwise_functions/isclose.cpp
Outdated
Show resolved
Hide resolved
fb66703
to
86fc82a
Compare
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 @vlad-perevezentsev for implementing the significant performance improvement. No more comments from me.
This PR suggests updating `dpnp.isclose()` function adding a scalar-specific SYCL kernels for both contiguous and stride cases to improve performance when `rtol` and `atol` are scalars. Also extends and updates tests for `dpnp.isclose()` The new kernel **improves performance** by **up to 10x** compared to the previous implementation when `rtol` and `atol` are scalars (tested on PVC). c9b9f70
This PR suggests updating
dpnp.isclose()
function adding a scalar-specific SYCL kernels for both contiguous and stride cases to improve performance whenrtol
andatol
are scalars.Also extends and updates tests for
dpnp.isclose()
The new kernel improves performance by up to 10x compared to the previous implementation when
rtol
andatol
are scalars (tested on PVC).CPU results:

GPU results:
