#4405: Move working logit sweeps to working folder

tenstorrent · Mar 22, 2024 · 39f7029 · 39f7029
1 parent d3349ea
commit 39f7029
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/...grayskull/pytorch_eltwise_logit_test.yaml → ...grayskull/pytorch_eltwise_logit_test.yaml b/...grayskull/pytorch_eltwise_logit_test.yaml → ...grayskull/pytorch_eltwise_logit_test.yaml
@@ -2,10 +2,11 @@ test-list:
   - eltwise-logit:
       shape:
         start-shape: [1, 1, 32, 32]
-        end-shape: [12, 24, 512, 512]
+        end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
         num-shapes: 1
         num-samples: 64
+        args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
@@ -21,14 +22,15 @@ test-list:
         data-layout: ["TILE"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-logit:
       shape:
         start-shape: [1, 1, 2, 2]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 1, 2]
         num-shapes: 1
         num-samples: 64
+        args-sampling-strategy: "all"
       env:
         # TT_PCI_DMA_BUF_SIZE: "1048576"
       datagen:
@@ -44,4 +46,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/.../wormhole/pytorch_eltwise_logit_test.yaml → .../wormhole/pytorch_eltwise_logit_test.yaml b/.../wormhole/pytorch_eltwise_logit_test.yaml → .../wormhole/pytorch_eltwise_logit_test.yaml
@@ -2,10 +2,11 @@ test-list:
   - eltwise-logit:
       shape:
         start-shape: [1, 1, 32, 32]
-        end-shape: [12, 24, 512, 512]
+        end-shape: [6, 12, 256, 256]
         interval: [1, 1, 32, 32]
         num-shapes: 1
         num-samples: 64
+        args-sampling-strategy: "all"
       datagen:
         function: gen_rand
         args:
@@ -19,14 +20,15 @@ test-list:
         data-layout: ["TILE"]
         data-type: ["BFLOAT16"]
         buffer-type: ["DRAM", "L1", "SYSTEM_MEMORY"]
-        out-buffer-type: ["DRAM"]
+        out-buffer-type: ["DRAM", "L1"]
   - eltwise-logit:
       shape:
         start-shape: [1, 1, 2, 2]
         end-shape: [6, 12, 256, 256]
         interval: [1, 1, 1, 2]
         num-shapes: 1
         num-samples: 64
+        args-sampling-strategy: "all"
       datagen:
         function: gen_rand
         args:
@@ -40,4 +42,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/...working_unit_tests/wormhole/test_logit.py → ...i_testing/unit_testing/misc/test_logit.py b/...working_unit_tests/wormhole/test_logit.py → ...i_testing/unit_testing/misc/test_logit.py