-
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
Feature/device manager #277
Feature/device manager #277
Conversation
8849425
to
e8a5068
Compare
This is not right: In [1]: import dpctl
In [2]: dpctl.SyclDevice("gpu")
Out[2]: <dpctl.SyclDevice at 0x7fd76edb44b0>
In [3]: dpctl.SyclQueue(dpctl.SyclDevice("gpu"))
Out[3]: <dpctl._sycl_queue.SyclQueue at 0x7fd7641299b0>
In [4]: Out[3].get_sycl_device() # UNEXPECTED: expecting to see SyclDevice, like in Out[2], but getting None
In [5]: !git status
On branch feature/device_selectors
Your branch is up to date with 'dipto/feature/device_selectors'.
nothing to commit, working tree clean
In [6]: !git log -1 --format=%H
28022afb8df19f46a7e488677dfb6f13c988c802 |
@oleksandr-pavlyk Needs more work. Something I did wrong results in |
@oleksandr-pavlyk I think this is good to go. Unfortunately the PR became way too huge than I had intended it to be, but will be great if you take a pass before we merge. |
99bc335
to
d7036d6
Compare
- Add a new dpctl_sycl_device_manager module to store a cache of root devices along with a cached context per device. - The device manager includes functions to get a vector of devices and the number of root devices of a specific type and backend combination. - Move the DPCTLDevice_DumpInfo() to a new module called dpctl_sycl_device_manager, and rename it to DPCTLDeviceMgr_PrintDeviceInfo(). - Introduce a new _sycl_device_factory module in the Python API. - Move device selector functions to the _sycl_device_factory module. - Add functions to get list of SYCL devies in dpctl.
d7036d6
to
a8e49c5
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.
Commits from this PR are also included in feature/queue_manager_v2
.
Closes #146
Closes #248
Closes #313
Various improvements to device management.
_sycl_device_factory
module to easily get the list of available devices in Python from SYCL.