#7292: Added dimenstion 3 sweeps for mse and l1 loss

tenstorrent · May 10, 2024 · 76a3f60 · 76a3f60
1 parent a8d7aa2
commit 76a3f60
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/..._testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml b/..._testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_l1_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"

diff --git a/...testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml b/...testing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_mse_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"

diff --git a/...i_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml b/...i_testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_l1_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"

diff --git a/..._testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml b/..._testing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_mse_loss_test.yaml
@@ -5,7 +5,7 @@ test-list:
         start-shape: [1, 1, 32, 32]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
-        num-dims: [2]
+        num-dims: [2,3]
         num-shapes: 2
         num-samples: 128
         args-sampling-strategy: "all"