#7292: TTNN MAJOR_ROW support for exp, acos, cos

tenstorrent · Apr 23, 2024 · 5c5c4b6 · 5c5c4b6
1 parent 94344d8
commit 5c5c4b6
Show file tree

Hide file tree

Showing 6 changed files with 158 additions and 8 deletions.
diff --git a/...ing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_acos_test.yaml b/...ing/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_acos_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -25,3 +25,28 @@ test-list:
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_acos_sweep.csv
+  - ttnn-acos:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: -1
+          high: 1
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_acos_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...ting/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_cos_test.yaml b/...ting/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_cos_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -25,3 +25,28 @@ test-list:
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_cos_sweep.csv
+  - ttnn-cos:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: 0
+          high: 6.283185307179586
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_cos_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...ting/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_exp_test.yaml b/...ting/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/ttnn_eltwise_exp_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -20,8 +20,33 @@ test-list:
         function: comp_pcc
       args-gen: gen_dtype_layout_device
       args:
-        data-layout: ["TILE", "ROW_MAJOR"]
+        data-layout: ["TILE"]
         data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_exp_sweep.csv
+  - ttnn-exp:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: -10
+          high: 10
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_exp_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...ting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_acos_test.yaml b/...ting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_acos_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -25,3 +25,28 @@ test-list:
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_acos_sweep.csv
+  - ttnn-acos:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: -1
+          high: 1
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_acos_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_cos_test.yaml b/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_cos_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -25,3 +25,28 @@ test-list:
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_cos_sweep.csv
+  - ttnn-cos:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: 0
+          high: 6.283185307179586
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_cos_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_exp_test.yaml b/...sting/sweep_tests/test_configs/ci_sweep_tests_working/wormhole/ttnn_eltwise_exp_test.yaml
@@ -7,7 +7,7 @@ test-list:
         interval: [1, 1, 32, 32]
         num-dims: [2, 3, 4]
         num-shapes: 1
-        num-samples: 128
+        num-samples: 64
         args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
@@ -20,8 +20,33 @@ test-list:
         function: comp_pcc
       args-gen: gen_dtype_layout_device
       args:
-        data-layout: ["TILE", "ROW_MAJOR"]
+        data-layout: ["TILE"]
         data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1"]
         out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_exp_sweep.csv
+  - ttnn-exp:
+      shape:
+        start-shape: [1, 1, 1, 2]
+        end-shape: [6, 12, 256, 256]
+        interval: [1, 1, 1, 2]
+        num-dims: [2, 3, 4]
+        num-shapes: 1
+        num-samples: 64
+        args-sampling-strategy: "all"
+      datagen:
+        function: gen_rand
+        args:
+          low: -10
+          high: 10
+      comparison:
+        function: comp_pcc
+      args-gen: gen_dtype_layout_device
+      output-file: eltwise_exp_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
+      args:
+        data-layout: ["ROW_MAJOR"]
+        data-type: ["BFLOAT16"]
+        buffer-type: ["DRAM", "L1"]
+        out-buffer-type: ["DRAM", "L1"]