offload_experiments: Update offload experiments

pulp-platform · May 15, 2024 · a8a372c · a8a372c
1 parent 26db8e2
commit a8a372c
Show file tree

Hide file tree

Showing 15 changed files with 2,070 additions and 685 deletions.
diff --git a/target/sim/cfg/M-Q1C4.hjson b/target/sim/cfg/M-Q1C4.hjson
diff --git a/target/sim/cfg/U-Q1C4.hjson b/target/sim/cfg/U-Q1C4.hjson
diff --git a/target/sim/cfg/U-Q2C4.hjson b/target/sim/cfg/U-Q2C4.hjson
diff --git a/target/sim/experiments/offload/ci.yml b/target/sim/experiments/offload/ci.yml
diff --git a/target/sim/experiments/offload/data/atax.json.tpl b/target/sim/experiments/offload/data/atax.json.tpl
@@ -3,6 +3,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 {
-    M: 16,
+    M: ${M},
     N: ${N}
 }
diff --git a/target/sim/experiments/offload/data/correlation.json.tpl b/target/sim/experiments/offload/data/correlation.json.tpl
@@ -3,6 +3,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 {
-    M: 16,
-    N: 8
+    M: ${M},
+    N: ${N}
 }
diff --git a/target/sim/experiments/offload/data/covariance.json.tpl b/target/sim/experiments/offload/data/covariance.json.tpl
@@ -3,6 +3,6 @@
 // SPDX-License-Identifier: Apache-2.0
 
 {
-    M: 16,
-    N: 8
+    M: ${M},
+    N: ${N}
 }
diff --git a/target/sim/experiments/offload/data/gemm.json.tpl b/target/sim/experiments/offload/data/gemm.json.tpl
@@ -3,18 +3,18 @@
 // SPDX-License-Identifier: SHL-0.51
 
 {
-    M: 256,
+    M: ${M},
     N: ${N},
     K: 4,
     beta: 0,
     ta: false,
     tb: true, // must be true for SIMD
     prec: "FP64",
     expand: 0,
-    m_tiles: 32, // number of tiles in M dimension
+    m_tiles: ${n_clusters_to_use}, // number of tiles in M dimension
     k_tiles: 1, // number of tiles in K dimension
     n_tiles: 1, // number of tiles in N dimension
     parallelize_k: 0,
     parallelize_m: 1,
-    baseline: false
+    baseline: true
 }