From b7eeb7e876495e43dabfd54a8d68f77b47e6520e Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 27 Aug 2025 21:31:56 +0200
Subject: [PATCH] python3Packages.torchao: 0.11.0 -> 0.12.0

Diff: https://github.com/pytorch/ao/compare/v0.11.0...v0.12.0

Changelog: https://github.com/pytorch/ao/releases/tag/v0.12.0
---
 pkgs/development/python-modules/torchao/default.nix | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/pkgs/development/python-modules/torchao/default.nix b/pkgs/development/python-modules/torchao/default.nix
index d249672428ab..8e6f65d31fbf 100644
--- a/pkgs/development/python-modules/torchao/default.nix
+++ b/pkgs/development/python-modules/torchao/default.nix
@@ -24,14 +24,14 @@
 
 buildPythonPackage rec {
   pname = "ao";
-  version = "0.11.0";
+  version = "0.12.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "ao";
     tag = "v${version}";
-    hash = "sha256-CNb9xaubOmIRanLq3TM4sBbszTcVK/WFpcq/sWpof44=";
+    hash = "sha256-J0aUce9Bu03Ff0ZjDKt39ZAX/UAif1S96SI7Gk4Hppw=";
   };
 
   build-system = [
@@ -75,6 +75,9 @@ buildPythonPackage rec {
     "test_gptq_mt"
   ]
   ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
+    # AssertionError: tensor(False) is not true
+    "test_quantize_per_token_cpu"
+
     # RuntimeError: failed to initialize QNNPACK
     "test_smooth_linear_cpu"