From a9ed21646a69c4d5b13f8c7e5c0e5e8da79c750c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E2=80=9CNenad?= <“npetrovic@tenstorrent.com”> Date: Thu, 16 May 2024 13:04:59 +0000 Subject: [PATCH] #7292: Fix op maps --- .../python_api_testing/sweep_tests/op_map.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/tests/ttnn/python_api_testing/sweep_tests/op_map.py b/tests/ttnn/python_api_testing/sweep_tests/op_map.py index c8730a9695b..2cfa3505f28 100644 --- a/tests/ttnn/python_api_testing/sweep_tests/op_map.py +++ b/tests/ttnn/python_api_testing/sweep_tests/op_map.py @@ -673,4 +673,20 @@ "tt_op": ttnn_ops.repeat, "pytorch_op": pytorch_ops.repeat_2, }, + "ttnn-eltwise-subtract_and_apply_activation": { + "tt_op": ttnn_ops.eltwise_subtract_and_apply_activation, + "pytorch_op": pytorch_ops.subtract_and_apply_activation, + }, + "ttnn-eltwise-subtract_and_apply_activation_": { + "tt_op": ttnn_ops.eltwise_subtract_and_apply_activation_, + "pytorch_op": pytorch_ops.subtract_and_apply_activation, + }, + "ttnn-eltwise-multiply_and_apply_activation": { + "tt_op": ttnn_ops.eltwise_multiply_and_apply_activation, + "pytorch_op": pytorch_ops.multiply_and_apply_activation, + }, + "ttnn-eltwise-multiply_and_apply_activation_": { + "tt_op": ttnn_ops.eltwise_multiply_and_apply_activation_, + "pytorch_op": pytorch_ops.multiply_and_apply_activation, + }, }