#7292: Fix op maps

tenstorrent · May 17, 2024 · a9ed216 · a9ed216
1 parent 0319be1
commit a9ed216
Showing 1 changed file with 16 additions and 0 deletions.
diff --git 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,
+    },
 }