Enable webgpu test in CI build #16983
snnn
started this conversation in
Show & Tell
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
This is the notes about how I enabled webgpu tests in onnxruntime CI build. It applies to both Azure DevOps pipelines and Github Actions.
Setup the build agent
First, the build agent must be run in an interactive mode. See https://learn.microsoft.com/en-us/azure/devops/pipelines/agents/agents?view=azure-devops&tabs=browser#interactive-or-service for more information.
Setup the cloud machine
Then, surely the machine needs to have a GPU device and do not forget to install GPU driver. Please note most Nvidia GPUs that can be found in data centers do not support video output and do not support DirectX. Since Chrome on Windows uses DirectX for rendering, your GPU must support DirectX. In Azure only the vm sizes that start with Standard_NV and Nvidia T4 based VM sizes can support DirectX. For example, NVadsA10 v5-series. However, NVadsA10 v5-series VMs cannot use the official CUDA drivers that are released through NVidia's official website, they need to use a special Azure Edition driver.
Launch the browser with special args
Add “--gpu-vendor-id=0x10de” and "--ignore-gpu-blocklist" to chrome's launch args, and do not do it in headless mode!
Both the args are needed. If without --ignore-gpu-blocklist, chrome will still create an extra software rendering device and use that one as the default GPU device.
Beta Was this translation helpful? Give feedback.
All reactions