From 1f7e90370971f4638ff44deeb3314aa83b774def Mon Sep 17 00:00:00 2001 From: mcw-anasuya Date: Mon, 21 Oct 2024 10:15:52 +0000 Subject: [PATCH] #8156: Add golden function --- tests/sweep_framework/sweeps/eltwise/unary/mish.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/sweep_framework/sweeps/eltwise/unary/mish.py b/tests/sweep_framework/sweeps/eltwise/unary/mish.py index 05f39ed6aee5..fc996ef658e0 100644 --- a/tests/sweep_framework/sweeps/eltwise/unary/mish.py +++ b/tests/sweep_framework/sweeps/eltwise/unary/mish.py @@ -6,7 +6,6 @@ from functools import partial import torch -import random import ttnn from tests.sweep_framework.sweep_utils.utils import gen_shapes from tests.tt_eager.python_api_testing.sweep_tests.generation_funcs import gen_func_with_cast_tt @@ -69,8 +68,8 @@ def run( partial(torch_random, low=-100, high=100, dtype=torch.float32), input_a_dtype )(input_shape) - mish = torch.nn.Mish() - torch_output_tensor = mish(torch_input_tensor_a) + golden_function = ttnn.get_golden_function(ttnn.mish) + torch_output_tensor = golden_function(torch_input_tensor_a) input_tensor_a = ttnn.from_torch( torch_input_tensor_a,