Skip to content

Commit

Permalink
#8555: Skip batch 8 resnet50 in ttnn because it's failing and being l…
Browse files Browse the repository at this point in the history
…ooked at (#8556)
  • Loading branch information
tt-rkim authored May 16, 2024
1 parent 2399c77 commit de9a19b
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -269,14 +269,16 @@ def create_test_infra(device, batch_size, act_dtype, weight_dtype, math_fidelity
@pytest.mark.parametrize(
"batch_size, act_dtype, weight_dtype, math_fidelity",
(
(8, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.LoFi), ## pass
(8, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.LoFi),
(16, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.HiFi2), ## pass
(16, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.LoFi), ## pass
(20, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.HiFi2), ## pass
(20, ttnn.bfloat8_b, ttnn.bfloat8_b, ttnn.MathFidelity.LoFi), ## pass
),
)
def test_resnet_50(device, batch_size, act_dtype, weight_dtype, math_fidelity):
if batch_size == 8:
pytest.skip("Failing, issue #8555")
ttnn.CONFIG.enable_logging = True
ttnn.CONFIG.enable_detailed_buffer_report = True
test_infra = create_test_infra(device, batch_size, act_dtype, weight_dtype, math_fidelity)
Expand Down

0 comments on commit de9a19b

Please sign in to comment.