-
Notifications
You must be signed in to change notification settings - Fork 30
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
Add sycl device sub group sizes property #985
Add sycl device sub group sizes property #985
Conversation
View rendered docs @ https://intelpython.github.io/dpctl/pulls/985/index.html |
Array API standard conformance tests for dpctl=0.14.0dev2=py310h8c27c75_4 ran successfully. |
Array API standard conformance tests for dpctl=0.14.0dev2=py310h8c27c75_5 ran successfully. |
6aaa41f
to
b7028f3
Compare
Array API standard conformance tests for dpctl=0.14.0dev2=py310h8c27c75_4 ran successfully. |
The function exposes `device::get_info<info::device::sub_group_sizes>()` which returns `std::vector<size_t>`. DPCTLDevice_GetSubGroupSizes returns pointer to allocated array, populated with the content of the result std::vector. res_len is set with the size of the result std::vector.
The property returns a vector of supported sub-group sizes. The relationship to `SyclDevice.max_num_sub_groups` is not clear at the moment.
Added a function that retrieves sub_group_sizes property of the device.
b7028f3
to
99f8a91
Compare
Deleted rendered PR docs from intelpython.github.com/dpctl, latest should be updated shortly. 🤞 |
Array API standard conformance tests for dpctl=0.14.0dev2=py310h8c27c75_9 ran successfully. |
1 similar comment
Array API standard conformance tests for dpctl=0.14.0dev2=py310h8c27c75_9 ran successfully. |
Thanks @oleksandr-pavlyk! |
This PR implements
dpctl.SyclDevice.sub_group_sizes
property as requested in #975