shapes -> workloads

Author: Jian Weng

apps/cpu/kernel/run.py

@@ -31,7 +31,7 @@
 
 #workloads = [(1, 288, 35, 35, 384, 288, 3, 3, 2, 2), (1, 160, 9, 9, 224, 160, 3, 3, 1, 1), (1, 1056, 7, 7, 192, 1056, 1, 1, 1, 1), (1, 80, 73, 73, 192, 80, 3, 3, 1, 1), (1, 128, 16, 16, 128, 128, 3, 3, 1, 1), (1, 192, 16, 16, 192, 192, 3, 3, 1, 1), (1, 256, 16, 16, 256, 256, 3, 3, 1, 1), (1, 1024, 14, 14, 512, 1024, 1, 1, 1, 1), (1, 128, 16, 16, 160, 128, 3, 3, 1, 1), (1, 576, 14, 14, 192, 576, 1, 1, 1, 1), (1, 96, 16, 16, 128, 96, 3, 3, 1, 1), (1, 1024, 14, 14, 256, 1024, 1, 1, 1, 1), (1, 576, 14, 14, 128, 576, 1, 1, 1, 1), (1, 64, 29, 29, 96, 64, 3, 3, 1, 1), (1, 64, 56, 56, 128, 64, 1, 1, 2, 2), (1, 608, 14, 14, 192, 608, 1, 1, 1, 1)]
 
-shapes = [
+workloads = [
         (1, 160, 9, 9, 224, 160, 3, 3, 1, 1),
         (1, 1056, 7, 7, 192, 1056, 1, 1, 1, 1),
         (1, 128, 16, 16, 128, 128, 3, 3, 1, 1),

apps/gpu/input

@@ -1 +1 @@
-1 608 14 14 192 608 1 1 1 1
+1 80 73 73 192 80 3 3 1 1

poc/vnni/run.py

@@ -14,7 +14,24 @@
 #(1024,15,15,2048,1024,1,1,2,2),
 #(128,65,65,256,128,3,3,2,2)]
 #
-workloads = [(1, 288, 35, 35, 384, 288, 3, 3, 2, 2), (1, 160, 9, 9, 224, 160, 3, 3, 1, 1), (1, 1056, 7, 7, 192, 1056, 1, 1, 1, 1), (1, 80, 73, 73, 192, 80, 3, 3, 1, 1), (1, 128, 16, 16, 128, 128, 3, 3, 1, 1), (1, 192, 16, 16, 192, 192, 3, 3, 1, 1), (1, 256, 16, 16, 256, 256, 3, 3, 1, 1), (1, 1024, 14, 14, 512, 1024, 1, 1, 1, 1), (1, 128, 16, 16, 160, 128, 3, 3, 1, 1), (1, 576, 14, 14, 192, 576, 1, 1, 1, 1), (1, 96, 16, 16, 128, 96, 3, 3, 1, 1), (1, 1024, 14, 14, 256, 1024, 1, 1, 1, 1), (1, 576, 14, 14, 128, 576, 1, 1, 1, 1), (1, 64, 29, 29, 96, 64, 3, 3, 1, 1), (1, 64, 56, 56, 128, 64, 1, 1, 2, 2), (1, 608, 14, 14, 192, 608, 1, 1, 1, 1)]
+workloads = [
+        (1, 160, 9, 9, 224, 160, 3, 3, 1, 1),
+        (1, 1056, 7, 7, 192, 1056, 1, 1, 1, 1),
+        (1, 128, 16, 16, 128, 128, 3, 3, 1, 1),
+        (1, 192, 16, 16, 192, 192, 3, 3, 1, 1),
+        (1, 256, 16, 16, 256, 256, 3, 3, 1, 1),
+        (1, 1024, 14, 14, 512, 1024, 1, 1, 1, 1),
+        (1, 128, 16, 16, 160, 128, 3, 3, 1, 1),
+        (1, 576, 14, 14, 192, 576, 1, 1, 1, 1),
+        (1, 96, 16, 16, 128, 96, 3, 3, 1, 1),
+        (1, 1024, 14, 14, 256, 1024, 1, 1, 1, 1),
+        (1, 576, 14, 14, 128, 576, 1, 1, 1, 1),
+        (1, 64, 29, 29, 96, 64, 3, 3, 1, 1),
+        (1, 64, 56, 56, 128, 64, 1, 1, 2, 2),
+        (1, 608, 14, 14, 192, 608, 1, 1, 1, 1),
+        (1, 288, 35, 35, 384, 288, 3, 3, 2, 2),
+        (1, 80, 73, 73, 192, 80, 3, 3, 1, 1),
+        ]
 #
 #workloads = [
 #(256, 16, 16, 256, 3, 1),