Skip to content

Commit

Permalink
#6365: Revert "#6365: Add ttnn host tests (#8210)" (#8879)
Browse files Browse the repository at this point in the history
because I'm an idiot who doesn't know how to be patient, clearly
will fix and merge another time

This reverts commit c3c844c.
  • Loading branch information
tt-rkim authored May 27, 2024
1 parent c3c844c commit 37ee1ab
Showing 1 changed file with 1 addition and 11 deletions.
12 changes: 1 addition & 11 deletions tests/end_to_end_tests/test_ttnn.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,19 +10,9 @@
import ttnn.operations.binary


@pytest.mark.eager_host_side
def test_ttnn_import_host(reset_seeds):
torch_input_tensor = torch.zeros(2, 4, dtype=torch.float32)

tensor = ttnn.from_torch(torch_input_tensor, dtype=ttnn.bfloat16)
torch_output_tensor = ttnn.to_torch(tensor)

assert torch.allclose(torch_input_tensor, torch_output_tensor)


@pytest.mark.eager_package_silicon
@pytest.mark.skip("Tries to open device twice")
def test_ttnn_import_sliicon(reset_seeds):
def test_ttnn_import(reset_seeds):
with ttnn.manage_device(device_id=0) as device:
pass

Expand Down

0 comments on commit 37ee1ab

Please sign in to comment.