Skip to content

Commit

Permalink
Added a comment
Browse files Browse the repository at this point in the history
  • Loading branch information
sminakov-tt committed Dec 11, 2024
1 parent 380adce commit 38ea1ee
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/ttnn/unit_tests/test_to_and_from_torch.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,5 +100,6 @@ def test_from_torch_large(device):
def test_to_for_01_rank(shape, layout, dtype):
torch_input_tensor = torch.rand(shape, dtype=dtype)
tensor = ttnn.from_torch(torch_input_tensor, layout=layout)
# Conversion in the opposite direction is not yet supported
# torch_output_tensor = ttnn.to_torch(tensor)
# assert torch.allclose(torch_input_tensor, torch_output_tensor)

0 comments on commit 38ea1ee

Please sign in to comment.