Add torch to origami pip modules (#5896)

* Add torch to origami pip modules

* Fix free space error

* Add numpy
This commit is contained in:
Ibrahim Wani
2026-01-29 09:56:31 -07:00
committed by GitHub
parent d1165b7359
commit 2da4c460ad

View File

@@ -47,8 +47,10 @@ parameters:
type: object
default:
- nanobind>=2.0.0
- numpy
- pytest
- pytest-cov
- torch
- name: rocmDependencies
type: object
default:
@@ -101,8 +103,7 @@ jobs:
- template: /.azuredevops/variables-global.yml
- name: ROCM_PATH
value: $(Agent.BuildDirectory)/rocm
pool:
vmImage: ${{ variables.BASE_BUILD_POOL }}
pool: ${{ variables.MEDIUM_BUILD_POOL }}
${{ if eq(job.os, 'almalinux8') }}:
container:
image: rocmexternalcicd.azurecr.io/manylinux228:latest