From 2da4c460addaa8c40ff7ed8d03fe5de519a52d87 Mon Sep 17 00:00:00 2001
From: Ibrahim Wani <113864060+ibrahimw1@users.noreply.github.com>
Date: Thu, 29 Jan 2026 09:56:31 -0700
Subject: [PATCH] Add torch to origami pip modules (#5896)

* Add torch to origami pip modules

* Fix free space error

* Add numpy
---
 .azuredevops/components/origami.yml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/.azuredevops/components/origami.yml b/.azuredevops/components/origami.yml
index 4bb5e4710..7e5d4178b 100644
--- a/.azuredevops/components/origami.yml
+++ b/.azuredevops/components/origami.yml
@@ -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