canonical / canonical/data-science-stack
Add GPU tests to run on Intel GPUs
Open
- Dominant language
- Python
- Stars
- 37
- Forks
- 10
- PR merge metrics
- No merged PRs in 30d
Description
The `integration-gpu` tests assume that the available GPUs are from NVIDIA.
https://github.com/canonical/data-science-stack/blob/2a10edcfc5280e8d481a8c8ddf645c80684cde37/tests/integration/test_dss.py#L70
Enabling testing on NVIDIA GPUs can be done by adding another `tox` environment like the one done for `integration-gpu`, but with a different notebook image:
https://github.com/canonical/data-science-stack/blob/2a10edcfc5280e8d481a8c8ddf645c80684cde37/tox.ini#L108
Furthermore, tests will have to be added to check using Intel GPUs.
Contributor guide
Assessment
This issue has not been assessed yet.