-
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 array API inspection utilities to dpctl.tensor
#1469
Conversation
This inspection utility is coming to the array API specification in the near future
View rendered docs @ https://intelpython.github.io/dpctl/pulls/1469/index.html |
Array API standard conformance tests for dpctl=0.15.1dev1=py310ha25a700_4 ran successfully. |
dpctl/tensor/_array_api.py
Outdated
def devices(self): | ||
return dpctl.get_devices() |
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.
It is possible to run in an environment where CUDA device is available, but dpctl
was not compiled to offload to NVidia GPUs.
Perhaps we should thin this list by ensuring that we can create an dpt.arange
for this device and cache that list.
It does not need to be done in this PR though
Array API standard conformance tests for dpctl=0.15.1dev1=py310ha25a700_6 ran successfully. |
The |
Disallows dtypes for `kind` kwarg in __array_namespace_info__().dtypes Removes `float16` from dtypes listed by __array_namespace_info__ as per spec Permits dpctl.tensor.Device objects in device keyword arguments in array API inspection utilities
13a4262
to
412692a
Compare
Array API standard conformance tests for dpctl=0.15.1dev1=py310ha25a700_16 ran successfully. |
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 @ndgrigorian
This pull request proposes implementing
dpctl.tensor.__array_namespace_info__
, an inspection utility coming soon to the array API specification.In addition, this PR adds an
__array_api_version__
attribute as per the specification and experimentally removes the--ci
flag from array API testing.