Add selu sweep (#15547)

Add new script for sweep: - selu
tenstorrent · Dec 13, 2024 · 414c895 · 414c895
1 parent 44e8301
commit 414c895
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/ttnn/unit_tests/operations/eltwise/test_eltwise_selu.py b/tests/ttnn/unit_tests/operations/eltwise/test_eltwise_selu.py
@@ -61,6 +61,22 @@ def run_eltwise_selu_tests(
         ttnn.L1_MEMORY_CONFIG,
         6411147,
     ),
+    (
+        [(3, 2, 191, 31)],
+        [ttnn.bfloat16],
+        [ttnn.TILE_LAYOUT],
+        [ttnn.DRAM_MEMORY_CONFIG],
+        ttnn.L1_MEMORY_CONFIG,
+        6861134,
+    ),
+    (
+        [(12, 225, 223)],
+        [ttnn.bfloat8_b],
+        [ttnn.TILE_LAYOUT],
+        [ttnn.DRAM_MEMORY_CONFIG],
+        ttnn.L1_MEMORY_CONFIG,
+        6411147,
+    ),
 ]