From eed9c165604a884f7100daf1f3173a474de2c09a Mon Sep 17 00:00:00 2001
From: Emma Qiao <qqiao@nvidia.com>
Date: Wed, 11 Feb 2026 17:40:40 +0800
Subject: [PATCH] [None][infra] Pin the torchao version (#11444)

Signed-off-by: qqiao <qqiao@nvidia.com>
---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 437a7470f7..0bada4a2d4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -77,7 +77,7 @@ partial_json_parser
 apache-tvm-ffi==0.1.6 # used for reduce nvidia-cutlass-dsl host overhead
 torch-c-dlpack-ext==0.1.3 # used for reduce nvidia-cutlass-dsl host overhead, optional package for improved torch tensor calling perf
 mistral-common==1.8.6
-torchao>=0.14.1
+torchao>=0.14.1,<0.16.0
 cuda-core
 llist
 cuda-tile>=1.0.1