-
Notifications
You must be signed in to change notification settings - Fork 180
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
FIX: prevent support_usm_ndarray
from changing queue if explicitly provided.
#1940
FIX: prevent support_usm_ndarray
from changing queue if explicitly provided.
#1940
Conversation
/intelci: run |
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 for founding out this issue.
This is critical for sklearnex.spmd ifaces with numpy + sycl_queue. Unfortunately we don't have properly testing yet for this API. Only some examples with dpnp/dpctl input, that not the case of this bug. Hopefully development on #1777 and #1909 will cover this cases as well.
I would like to have some tests in case. I can follow this add it here.
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.
Generally looks good to me. @ethanglaser please take a look as well. This is critical for spmd ifaces with numpy + sycl queue inputs.
1b35d8b
to
dbe7e69
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.
We went a little bit wrong in our thoughts, but it seems like it should be normal now.
Again without testing this primitive we could possible have some other issues. It would be great to add tests to check for this estimator expected behavior for the dummy function. Could be done in separate PR as follow up, but make sure that we have ticket covered. Thank you!
a676415
to
6252d93
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.
LGTM. Assuming green CI
6252d93
to
8403a00
Compare
Please run black formatting on the file |
8403a00
to
c4aacc7
Compare
/intelci: run |
@mergify backport rls/2024.6.0-rls |
✅ Backports have been created
|
Description
support_usm_ndarray
. Now it does not change the queue if it was explicitly provided to args. Previous logic does not allow to usenp.array
+SyclQueue
interface with decorated method.queue_param
is removed fromsupport_usm_ndarray
since the check can be done via kwargs checking.support_usm_ndarray
with highlighting the change