summaryrefslogtreecommitdiff
path: root/TRNS/dimes-hetsim-hbm.sh
diff options
context:
space:
mode:
Diffstat (limited to 'TRNS/dimes-hetsim-hbm.sh')
-rwxr-xr-xTRNS/dimes-hetsim-hbm.sh30
1 files changed, 25 insertions, 5 deletions
diff --git a/TRNS/dimes-hetsim-hbm.sh b/TRNS/dimes-hetsim-hbm.sh
index a7240a9..b6d645a 100755
--- a/TRNS/dimes-hetsim-hbm.sh
+++ b/TRNS/dimes-hetsim-hbm.sh
@@ -6,19 +6,38 @@ make -B NUMA=1
mkdir -p log/$(hostname)
fn=log/$(hostname)/$(date +%Y%m%d)
-# DPU version: -m m -n n -o M_ -p N_
-# input size: sizeof(T) * M_ * m * N_ * n
+# Args: -m m -n n -o M_ -p N_
+#
+# Input: (M_ * m) × (N_ * n) matrix
+# Output: (N_* n) × (M_ * m) matrix
+# Step 1: transpose (M_ * m) × N_ matrix that consits of tiles of size n
+# CPU version: explicit
+# DPU version: implicit (appropriate write operations to DPUs)
+# Step 2: transpose m × n matrix; this happens N_ * M_ times.
+# DPU version: Each tasklet transposes a single m × n matrix / tile.
+# (16 × 8 tile takes up 1 KiB WRAM)
+# Step 3: Transpose M_ × n matrix that consists of tiles of size m.
+#
+# Note for DPU version: if M_ > #DPUs, steps 1 through 3 are repeated.
+# Number of repetitions == ceil(M_ / #DPUS)
+# For Hetsim benchmarks, we set M_ == #DPUs to simplify the task graph (no repetitions that depend on the number of available DPUs).
+# Just in case, there is also a configuration with M_ == 2048 independent of #DPUs
+#
+# input size: uint64(DPU)/double(CPU) * M_ * m * N_ * n
+# output size: uint64(DPU)/double(CPU) * M_ * m * N_ * n -- on DPU only; CPU version operates in-place
# Upstream DPU version uses int64_t, -p 2048 -o 12288 -x 1 [implicit -m 16 -n 8]
# Upstream CPU version uses double, -p 2556 -o 4096 -m 16 -n 8 and fails with -o 12288 (allocation error)
#
-# Here: -p 2048 -o 2048 -m 16 -n 8 for both
+# -p 2048 -o 2048 -m 16 -n 8 -> matrix size: 4 GiB
+# -p [64 .. 2304] -o 2048 -m 16 -n 8 -> matrix size: 128 MiB .. 4.5 GiB
(
echo "CPU single-node operation (1/2)" >&2
parallel -j1 --eta --joblog ${fn}.1.joblog --header : \
- ./trns -w 0 -r 40 -p 2048 -o 2048 -m 16 -n 8 -t {nr_threads} -a {ram} -b {ram} -c {cpu} \
+ ./trns -w 0 -r 5 -p {p} -o 2048 -m 16 -n 8 -t {nr_threads} -a {ram} -b {ram} -c {cpu} \
+ ::: p 64 128 256 512 768 1024 1536 2048 2304 \
::: ram $(seq 0 15) \
::: cpu $(seq 0 7) \
::: nr_threads 1 2 4 8 12 16
@@ -26,7 +45,8 @@ parallel -j1 --eta --joblog ${fn}.1.joblog --header : \
echo "CPU multi-node operation (1/2)" >&2
parallel -j1 --eta --joblog ${fn}.2.joblog --header : \
- ./trns -w 0 -r 40 -p 2048 -o 2048 -m 16 -n 8 -t {nr_threads} -a {ram} -b {ram} -c {cpu} \
+ ./trns -w 0 -r 40 -p {p} -o 2048 -m 16 -n 8 -t {nr_threads} -a {ram} -b {ram} -c {cpu} \
+ ::: p 64 128 256 512 768 1024 1536 2048 2304 \
::: ram $(seq 0 15) \
::: cpu -1 \
::: nr_threads 32 48 64 96 128