Skip to content
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 GPU tests to run on Intel GPUs #189

Open
motjuste opened this issue Nov 29, 2024 · 1 comment
Open

Add GPU tests to run on Intel GPUs #189

motjuste opened this issue Nov 29, 2024 · 1 comment

Comments

@motjuste
Copy link
Contributor

The integration-gpu tests assume that the available GPUs are from NVIDIA.

) # This checks for the presence of NVIDIA in the log, assuming only NVIDIA GPUs are used.

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:

pytest {[vars]tst_path}/integration/ -m gpu -v --tb native -s --notebook-image={env:NOTEBOOK_IMAGE:kubeflownotebookswg/jupyter-pytorch-cuda-full:v1.8.0} {posargs}

Furthermore, tests will have to be added to check using Intel GPUs.

@misohu
Copy link
Member

misohu commented Dec 2, 2024

Hi @motjuste sounds like a good idea but the problem is we dont have hardware to develop the tests with Intel. I know we can use testflinger to run a machine in the lab but right now seems like a too much effort.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants