Skip to content

Commit

Permalink
#14342: Hotfix skip grayskull test on bernoulli op (#14343)
Browse files Browse the repository at this point in the history
#14342: skip grayskull test on bernoulli op
  • Loading branch information
BuiChiTrung authored Oct 28, 2024
1 parent 0911990 commit c5e11ac
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/ttnn/unit_tests/operations/test_bernoulli.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
compute_kernel_ids,
get_lib_dtype,
)
from models.utility_functions import skip_for_grayskull
from collections import Counter
from loguru import logger

Expand Down Expand Up @@ -55,6 +56,7 @@ def run_bernoulli(shape, in_dtype, out_dtype, device, is_out_alloc=False, comput


# fmt: off
@skip_for_grayskull("Requires wormhole_b0 to run")
@pytest.mark.parametrize("shape",
[
[2003],
Expand Down Expand Up @@ -86,6 +88,7 @@ def test_bernoulli(shape, in_dtype, out_dtype, device, is_out_alloc):
run_bernoulli(shape, in_dtype, out_dtype, device, is_out_alloc)


@skip_for_grayskull("Requires wormhole_b0 to run")
@pytest.mark.parametrize(
"shape",
[
Expand All @@ -108,6 +111,7 @@ def test_bernoulli_callback(shape, in_dtype, out_dtype, device, is_out_alloc, us
assert num_program_cache_entries_list[0] == num_program_cache_entries_list[1]


@skip_for_grayskull("Requires wormhole_b0 to run")
@pytest.mark.parametrize(
"shape",
[[512, 512], [5, 4, 70, 40]],
Expand Down

0 comments on commit c5e11ac

Please sign in to comment.