Skip to content

Commit

Permalink
#0: Disable ttlib resnet batch8
Browse files Browse the repository at this point in the history
This variant was failing, but we are transitioning to the ttnn version
of this model.  For now we opted to just remove this config since this
test is deprecated and will eventually be removed.
  • Loading branch information
nsmithtt committed Apr 17, 2024
1 parent d755048 commit 9449087
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/demos/resnet/tests/test_metal_resnet50.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@


@skip_for_wormhole_b0("This test is not supported on WHB0, please use the TTNN version.")
@pytest.mark.parametrize("batch_size", [1, 2, 8, 16, 20], ids=["batch_1", "batch_2", "batch_8", "batch_16", "batch_20"])
@pytest.mark.parametrize("batch_size", [1, 2, 16, 20], ids=["batch_1", "batch_2", "batch_16", "batch_20"])
@pytest.mark.parametrize(
"weights_dtype",
[tt_lib.tensor.DataType.BFLOAT16, tt_lib.tensor.DataType.BFLOAT8_B],
Expand Down

0 comments on commit 9449087

Please sign in to comment.