change transpose impl

jerryzh168 · jerryzh168 · commit 713946690797 · 2024-09-25T18:53:15.000-07:00
diff --git a/torchao/dtypes/affine_quantized_tensor.py b/torchao/dtypes/affine_quantized_tensor.py
@@ -603,7 +603,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs):
         elif func is aten.t.default:
             tensor = args[0]
             new = tensor.__class__(
-                tensor.int_data.view(tensor.shape[::-1]), tensor.scale, tensor.zero_point, tensor.layout_type
+                tensor.int_data.t(), tensor.scale, tensor.zero_point, tensor.layout_type
             )
             return return_and_correct_aliasing(func, args, kwargs, new)
 

Original file line number	Diff line number	Diff line change
`@@ -603,7 +603,7 @@ def __torch_dispatch__(cls, func, types, args, kwargs):`
`603`	`603`	`elif func is aten.t.default:`
`604`	`604`	`tensor = args[0]`
`605`	`605`	`new = tensor.__class__(`
`606`		`- tensor.int_data.view(tensor.shape[::-1]), tensor.scale, tensor.zero_point, tensor.layout_type`
	`606`	`+ tensor.int_data.t(), tensor.scale, tensor.zero_point, tensor.layout_type`
`607`	`607`	`)`
`608`	`608`	`return return_and_correct_aliasing(func, args, kwargs, new)`
`609`	`609`