From b83222d873eb5359d1f9ff841d51556a20c2cb88 Mon Sep 17 00:00:00 2001 From: Daniel Vega-Myhre Date: Fri, 17 Oct 2025 18:08:08 -0700 Subject: [PATCH] fix AQT test --- test/dtypes/test_affine_quantized.py | 2 +- torchao/dtypes/affine_quantized_tensor_ops.py | 2 +- torchao/dtypes/floatx/cutlass_semi_sparse_layout.py | 2 +- torchao/dtypes/uintx/cutlass_int4_packed_layout.py | 2 +- torchao/dtypes/uintx/gemlite_layout.py | 2 +- torchao/dtypes/uintx/plain_layout.py | 2 +- torchao/dtypes/uintx/q_dq_layout.py | 2 +- torchao/dtypes/uintx/tensor_core_tiled_layout.py | 2 +- torchao/quantization/linear_activation_quantized_tensor.py | 2 +- 9 files changed, 9 insertions(+), 9 deletions(-) diff --git a/test/dtypes/test_affine_quantized.py b/test/dtypes/test_affine_quantized.py index 83f32c8420..9679f6975a 100644 --- a/test/dtypes/test_affine_quantized.py +++ b/test/dtypes/test_affine_quantized.py @@ -275,7 +275,7 @@ def test_copy__mismatch_metadata(self, apply_quant): # copy should fail due to shape mismatch with self.assertRaisesRegex( - ValueError, "Not supported args for copy_ due to metadata mistach:" + ValueError, "Not supported args for copy_ due to metadata mismatch:" ): ql2.weight.copy_(ql.weight) diff --git a/torchao/dtypes/affine_quantized_tensor_ops.py b/torchao/dtypes/affine_quantized_tensor_ops.py index 7d693451cd..ffadece729 100644 --- a/torchao/dtypes/affine_quantized_tensor_ops.py +++ b/torchao/dtypes/affine_quantized_tensor_ops.py @@ -462,7 +462,7 @@ def _(func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) diff --git a/torchao/dtypes/floatx/cutlass_semi_sparse_layout.py b/torchao/dtypes/floatx/cutlass_semi_sparse_layout.py index 1787d18672..263ad2e699 100644 --- a/torchao/dtypes/floatx/cutlass_semi_sparse_layout.py +++ b/torchao/dtypes/floatx/cutlass_semi_sparse_layout.py @@ -98,7 +98,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) elif func is aten.clone.default: return return_and_correct_aliasing( diff --git a/torchao/dtypes/uintx/cutlass_int4_packed_layout.py b/torchao/dtypes/uintx/cutlass_int4_packed_layout.py index d2fcaf9045..d680f4cf77 100644 --- a/torchao/dtypes/uintx/cutlass_int4_packed_layout.py +++ b/torchao/dtypes/uintx/cutlass_int4_packed_layout.py @@ -102,7 +102,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) raise NotImplementedError( diff --git a/torchao/dtypes/uintx/gemlite_layout.py b/torchao/dtypes/uintx/gemlite_layout.py index 51b453de8a..8a8f2309c9 100644 --- a/torchao/dtypes/uintx/gemlite_layout.py +++ b/torchao/dtypes/uintx/gemlite_layout.py @@ -406,7 +406,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): self.gemlite_kwargs[key] = src.gemlite_kwargs[key] return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) raise NotImplementedError( diff --git a/torchao/dtypes/uintx/plain_layout.py b/torchao/dtypes/uintx/plain_layout.py index 3551214d7e..1e9b57e130 100644 --- a/torchao/dtypes/uintx/plain_layout.py +++ b/torchao/dtypes/uintx/plain_layout.py @@ -144,7 +144,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) elif func is aten.t.default: diff --git a/torchao/dtypes/uintx/q_dq_layout.py b/torchao/dtypes/uintx/q_dq_layout.py index be2c7fe16c..4c043d9b03 100644 --- a/torchao/dtypes/uintx/q_dq_layout.py +++ b/torchao/dtypes/uintx/q_dq_layout.py @@ -163,7 +163,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) elif func is aten.t.default: diff --git a/torchao/dtypes/uintx/tensor_core_tiled_layout.py b/torchao/dtypes/uintx/tensor_core_tiled_layout.py index 1961cc33c5..625f4c8fb9 100644 --- a/torchao/dtypes/uintx/tensor_core_tiled_layout.py +++ b/torchao/dtypes/uintx/tensor_core_tiled_layout.py @@ -361,7 +361,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs): getattr(self, tensor_name).copy_(getattr(src, tensor_name)) return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" ) if func in [aten.select.int, aten.index.Tensor]: diff --git a/torchao/quantization/linear_activation_quantized_tensor.py b/torchao/quantization/linear_activation_quantized_tensor.py index 40836bd920..01cc9a424d 100644 --- a/torchao/quantization/linear_activation_quantized_tensor.py +++ b/torchao/quantization/linear_activation_quantized_tensor.py @@ -226,7 +226,7 @@ def _(func, types, args, kwargs): return raise ValueError( - f"Not supported args for copy_ due to metadata mistach: {args[0], args[1]}" + f"Not supported args for copy_ due to metadata mismatch: {args[0], args[1]}" )