From ad4db928dd3ab24cc97ebbfa6122a5f6299bb369 Mon Sep 17 00:00:00 2001 From: Su Tong Date: Mon, 15 Dec 2025 14:41:48 +0800 Subject: [PATCH 1/2] fix the transpose error --- .../workflows/float8/float8_tensor.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/torchao/quantization/quantize_/workflows/float8/float8_tensor.py b/torchao/quantization/quantize_/workflows/float8/float8_tensor.py index 5fe1555f2a..86f072eacd 100644 --- a/torchao/quantization/quantize_/workflows/float8/float8_tensor.py +++ b/torchao/quantization/quantize_/workflows/float8/float8_tensor.py @@ -965,6 +965,24 @@ def _(func, types, args, kwargs): return return_and_correct_aliasing(func, args, kwargs, new_tensor) +@implements_torch_function(torch.Tensor.t) +def _(func, types, args, kwargs): + assert len(args) == 1 + self = args[0] + assert len(self.block_size) == 2 + new_tensor = self.__class__( + self.qdata.t(), + self.scale.t(), + (self.block_size[1], self.block_size[0]), + self.mm_config, + self.act_quant_kwargs, + self.kernel_preference, + self.dtype, + ) + return new_tensor + + + @implements(aten.split.Tensor) def _(func, types, args, kwargs): tensor, split_size_or_sections, dim = args From ab582641a1d80b2e791627278f5ab3ed932e5d6b Mon Sep 17 00:00:00 2001 From: Su Tong Date: Mon, 15 Dec 2025 14:49:56 +0800 Subject: [PATCH 2/2] Update float8_tensor.py --- torchao/quantization/quantize_/workflows/float8/float8_tensor.py | 1 - 1 file changed, 1 deletion(-) diff --git a/torchao/quantization/quantize_/workflows/float8/float8_tensor.py b/torchao/quantization/quantize_/workflows/float8/float8_tensor.py index 86f072eacd..7aca6f0a7e 100644 --- a/torchao/quantization/quantize_/workflows/float8/float8_tensor.py +++ b/torchao/quantization/quantize_/workflows/float8/float8_tensor.py @@ -982,7 +982,6 @@ def _(func, types, args, kwargs): return new_tensor - @implements(aten.split.Tensor) def _(func, types, args, kwargs): tensor, split_size_or_sections, dim = args