From c5e11acd46e5d4a1cc7a6e385d7e473aea5acd68 Mon Sep 17 00:00:00 2001
From: Bui Chi Trung <52347285+BuiChiTrung@users.noreply.github.com>
Date: Mon, 28 Oct 2024 18:26:52 +0700
Subject: [PATCH] #14342: Hotfix skip grayskull test on bernoulli op (#14343)

#14342: skip grayskull test on bernoulli op
---
 tests/ttnn/unit_tests/operations/test_bernoulli.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tests/ttnn/unit_tests/operations/test_bernoulli.py b/tests/ttnn/unit_tests/operations/test_bernoulli.py
index 5c3c68ab5c4..104259801b9 100644
--- a/tests/ttnn/unit_tests/operations/test_bernoulli.py
+++ b/tests/ttnn/unit_tests/operations/test_bernoulli.py
@@ -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
 
@@ -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],
@@ -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",
     [
@@ -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]],