flan/winnow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: add platform markers to GPU extras, use --extra cpu in CI, simplify lockfile workflow

Browse Source 
- rocm/intel/gpu extras are x86_64-only; aarch64 wheels don't exist so uv
  failed to resolve them when required-environments includes aarch64
- test.yml: switch from --all-extras (broken by conflicts + missing wheels)
  to --extra cpu which is cross-platform and sufficient for unit tests
- update-lockfile.yml: drop old file-swap loop; single pyproject means a
  single uv lock run and a single uv.lock to commit
This commit is contained in:
flan
2026-06-14 20:04:25 +00:00
parent e2a1924fb0
commit 02c56493f6
3 changed files with 14 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyproject.toml
+6 -3
View File
@@ -26,9 +26,12 @@ dependencies = [
]
[project.optional-dependencies]
gpu = ["onnxruntime-gpu>=1.23.2", "nvidia-cudnn-cu12>=9.0.0"]
rocm = ["onnxruntime-rocm>=1.16.0"]
intel = ["onnxruntime-openvino>=1.20.0"]
gpu = [
"onnxruntime-gpu>=1.23.2; sys_platform == 'linux' and platform_machine == 'x86_64'",
"nvidia-cudnn-cu12>=9.0.0; sys_platform == 'linux' and platform_machine == 'x86_64'",
]
rocm = ["onnxruntime-rocm>=1.16.0; sys_platform == 'linux' and platform_machine == 'x86_64'"]
intel = ["onnxruntime-openvino>=1.20.0; sys_platform == 'linux' and platform_machine == 'x86_64'"]
cpu = ["onnxruntime>=1.23.2"]
[project.scripts]