mirror of
https://github.com/ROCm/ROCm.git
synced 2026-01-31 17:38:41 -05:00
Add torch to origami pip modules (#5896)
* Add torch to origami pip modules * Fix free space error * Add numpy
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user