rename amd flavor to rocm

This commit is contained in:
mauwii 2023-02-24 06:20:44 +01:00
parent b7718985d5
commit 01f8c37bd3
No known key found for this signature in database
GPG Key ID: D923DB04ADB3F5AB

View File

@ -26,11 +26,11 @@ jobs:
fail-fast: false
matrix:
flavor:
- amd
- rocm
- cuda
- cpu
include:
- flavor: amd
- flavor: rocm
pip-extra-index-url: 'https://download.pytorch.org/whl/rocm5.2'
- flavor: cuda
pip-extra-index-url: ''