#6586: Add bfloat8_b and L1 for GS eqz, gte, sigmoid and other test s…

…weeps
tenstorrent · Apr 1, 2024 · 7f3c091 · 7f3c091
1 parent c03525e
commit 7f3c091
Show file tree

Hide file tree

Showing 16 changed files with 78 additions and 108 deletions.
diff --git a/.../sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_add1_test.yaml b/.../sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_add1_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -22,7 +20,7 @@ test-list:
         data-layout: ["TILE"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_add1_sweep.csv
   - eltwise-add1:
       shape:
@@ -32,8 +30,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -46,5 +42,5 @@ test-list:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_add1_sweep.csv
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_eqz_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_eqz_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -20,10 +18,12 @@ test-list:
       args-gen: gen_dtype_layout_device
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_eqz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
   - eltwise-eqz:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -46,5 +44,7 @@ test-list:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_eqz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gez_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gez_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -20,10 +18,12 @@ test-list:
       args-gen: gen_dtype_layout_device
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_gez_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
   - eltwise-gez:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -46,5 +44,7 @@ test-list:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_gez_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
diff --git a/...ng/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gt_test.yaml b/...ng/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gt_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 2
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -20,10 +18,12 @@ test-list:
       args-gen: gen_dtype_layout_device
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_gt_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
   - eltwise-gt:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 2
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -46,5 +44,7 @@ test-list:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
       output-file: eltwise_gt_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gte_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gte_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 2
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -19,11 +17,13 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_gte_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-gte:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 2
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -43,8 +41,10 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_gte_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gtz_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_gtz_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -19,11 +17,13 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_gtz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["TILE"]
         data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-gtz:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -43,8 +41,10 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_gtz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_hardsigmoid_test.yaml b/...tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_hardsigmoid_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -23,7 +21,7 @@ test-list:
         data-layout: ["TILE"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-hardsigmoid:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +30,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -47,4 +43,4 @@ test-list:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_lez_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_lez_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -19,11 +17,13 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_lez_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-lez:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -43,8 +41,10 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_lez_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
diff --git a/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_ltz_test.yaml b/...g/sweep_tests/test_configs/ci_sweep_tests_working/grayskull/pytorch_eltwise_ltz_test.yaml
@@ -8,8 +8,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -19,11 +17,13 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_ltz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["TILE"]
-        data-type: ["BFLOAT16"]
+        data-type: ["BFLOAT16", "BFLOAT8_B"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-ltz:
       shape:
         start-shape: [1, 1, 2, 2]
@@ -32,8 +32,6 @@ test-list:
         num-shapes: 1
         num-samples: 64
         args-sampling-strategy: "all"
-      env:
-        # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
         function: gen_rand
         args:
@@ -43,8 +41,10 @@ test-list:
         function: comp_equal
       args-gen: gen_dtype_layout_device
       output-file: eltwise_ltz_sweep.csv
+      env:
+        # TT_PCI_DMA_BUF_SIZE: "1048576"
       args:
         data-layout: ["ROW_MAJOR"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]