#5589: Sub YAML files modification

tenstorrent · Mar 14, 2024 · fa91e6d · fa91e6d
1 parent 3e799de
commit fa91e6d
Show file tree

Hide file tree

Showing 9 changed files with 163 additions and 109 deletions.
diff --git a/...sting/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/ttnn_eltwise_sub_test.yaml b/...sting/sweep_tests/test_configs/ci_sweep_tests_broken/grayskull/ttnn_eltwise_sub_test.yaml
diff --git a/...esting/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/ttnn_eltwise_sub_test.yaml b/...esting/sweep_tests/test_configs/ci_sweep_tests_broken/wormhole/ttnn_eltwise_sub_test.yaml
diff --git a/...ng/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_ldexp_test.yaml b/...ng/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_ldexp_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-ldexp:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -10
+          high: 10
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_eltwise_ldexp_sweep.csv
diff --git a/...eep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp2_test.yaml b/...eep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp2_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-logaddexp2:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -64
+          high: 64
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_logaddexp2_sweep.csv
diff --git a/...weep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp_test.yaml b/...weep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_logaddexp_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-logaddexp:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -64
+          high: 64
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_logaddexp_sweep.csv
diff --git a/...ing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_ldexp_test.yaml b/...ing/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_ldexp_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-ldexp:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -10
+          high: 10
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_eltwise_ldexp_sweep.csv
diff --git a/...weep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp2_test.yaml b/...weep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp2_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-logaddexp2:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -10
+          high: 10
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_logaddexp2_sweep.csv
diff --git a/...sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp_test.yaml b/...sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_logaddexp_test.yaml
@@ -0,0 +1,27 @@
+---
+test-list:
+  - ttnn-logaddexp:
+      shape:
+        start-shape: [1, 1, 32, 32]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 32, 32]
+        num-dims: [2, 3, 4]
+        num-shapes: 2
+        num-samples: 128
+        args-sampling-strategy: "all"
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      datagen:
+        function: gen_rand
+        args:
+          low: -64
+          high: 64
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      args:
+        data-layout: ["TILE"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
+      output-file: eltwise_logaddexp_sweep.csv
diff --git a/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_sub_test.yaml b/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_sub_test.yaml
@@ -21,7 +21,7 @@ test-list:
       args-gen: gen_scalar_args
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16", "BFLOAT8_B"]
+        data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_sub_sweep.csv