Skip to content

Commit 401ea8e

Browse files
authored
Skip np.transpose for identity dim_order in process_node
Differential Revision: D101430550 Pull Request resolved: pytorch#18977
1 parent 0af4221 commit 401ea8e

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

backends/arm/process_node.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,8 @@ def _tensor_to_numpy_with_dim_order(
4242
np_tensor = tensor.view(torch.uint16).numpy().view(ml_dtypes.bfloat16)
4343
else:
4444
np_tensor = tensor.numpy()
45+
if dim_order == tuple(range(len(dim_order))):
46+
return np_tensor
4547
return np.transpose(np_tensor, dim_order)
4648

4749

0 commit comments

Comments
 (0)