4 files changed, 151 insertions, 0 deletions
diff --git a/TRNS/include/common.h b/TRNS/include/common.h
new file mode 100644
index 0000000..6a94c62
--- /dev/null
+++ b/TRNS/include/common.h
@@ -0,0 +1,42 @@
+#ifndef _COMMON_H_
+#define _COMMON_H_
+
+// Transfer size between MRAM and WRAM
+#ifdef BL
+#define BLOCK_SIZE_LOG2 BL
+#define BLOCK_SIZE (1 << BLOCK_SIZE_LOG2)
+#else
+#define BLOCK_SIZE_LOG2 8
+#define BLOCK_SIZE (1 << BLOCK_SIZE_LOG2)
+#define BL BLOCK_SIZE_LOG2
+#endif
+
+// Data type
+#define T int64_t
+
+enum kernels {
+	kernel1 = 0,
+	kernel2 = 1,
+	nr_kernels = 2,
+};
+
+// Structures used by both the host and the dpu to communicate information 
+typedef struct {
+    uint32_t m;
+    uint32_t n;
+    uint32_t M_;
+	enum kernels kernel;
+} dpu_arguments_t;
+
+#ifndef ENERGY
+#define ENERGY 0
+#endif
+#define PRINT 0 
+
+#define ANSI_COLOR_RED     "\x1b[31m"
+#define ANSI_COLOR_GREEN   "\x1b[32m"
+#define ANSI_COLOR_RESET   "\x1b[0m"
+
+#define divceil(n, m) (((n)-1) / (m) + 1)
+#define roundup(n, m) ((n / m) * m + m)
+#endif
diff --git a/TRNS/include/dfatool_host.ah b/TRNS/include/dfatool_host.ah
new file mode 100644
index 0000000..72978cc
--- /dev/null
+++ b/TRNS/include/dfatool_host.ah
@@ -0,0 +1,36 @@
+#pragma once
+
+#include <sys/time.h>
+#include "dfatool_host_dpu.ah"
+
+aspect DfatoolHostTiming : public DfatoolHostDPUTiming {
+
+	unsigned int n_rows_outer, n_rows_tile, n_cols_outer, n_cols_tile;
+	unsigned int element_size;
+
+	virtual int getKernel() { return kernel; }
+
+	DfatoolHostTiming() {
+		element_size = sizeof(T);
+	}
+
+	advice call("% input_params(...)") : after() {
+		Params* p = tjp->result();
+		/*
+		 * Input: (M_ * m) × (N_ * n) matrix
+		 */
+		n_rows_outer = p->M_;
+		n_rows_tile = p->m;
+		n_cols_outer = p->N_;
+		n_cols_tile = p->n;
+		printf("[>>] TRNS | n_dpus=%u n_rows_outer=%u n_rows_tile=%u n_cols_outer=%u n_cols_tile=%u\n", NR_DPUS, n_rows_outer, n_rows_tile, n_cols_outer, n_cols_tile);
+	}
+
+	advice call("% trns_host(...)") : after() {
+		printf("[--] TRNS | n_dpus=%u n_rows_outer=%u n_rows_tile=%u n_cols_outer=%u n_cols_tile=%u\n", NR_DPUS, n_rows_outer, n_rows_tile, n_cols_outer, n_cols_tile);
+	}
+
+	advice execution("% main(...)") : after() {
+		printf("[<<] TRNS | n_dpus=%u n_rows_outer=%u n_rows_tile=%u n_cols_outer=%u n_cols_tile=%u\n", NR_DPUS, n_rows_outer, n_rows_tile, n_cols_outer, n_cols_tile);
+	}
+};
diff --git a/TRNS/include/params.h b/TRNS/include/params.h
new file mode 100644
index 0000000..385490e
--- /dev/null
+++ b/TRNS/include/params.h
@@ -0,0 +1,68 @@
+#ifndef _PARAMS_H_
+#define _PARAMS_H_
+
+#include "common.h"
+
+typedef struct Params {
+    unsigned int   M_;
+    unsigned int   m;
+    unsigned int   N_;
+    unsigned int   n;
+    int   n_warmup;
+    int   n_reps;
+    int  exp;
+}Params;
+
+static void usage() {
+    fprintf(stderr,
+        "\nUsage:  ./program [options]"
+        "\n"
+        "\nGeneral options:"
+        "\n    -h        help"
+        "\n    -w <W>    # of untimed warmup iterations (default=1)"
+        "\n    -e <E>    # of timed repetition iterations (default=3)"
+        "\n    -x <X>    Weak (0) or strong (1) scaling (default=1)"
+        "\n"
+        "\nBenchmark-specific options:"
+        "\n    -m <I>    m (default=16 elements)"
+        "\n    -n <I>    n (default=8 elements)"
+        "\n    -o <I>    M_ (default=12288 elements)"
+        "\n    -p <I>    N_ (default=1 elements)"
+        "\n");
+}
+
+struct Params input_params(int argc, char **argv) {
+    struct Params p;
+    p.M_            = 12288;
+    p.m             = 16;
+    p.N_            = 1;
+    p.n             = 8;
+    p.n_warmup      = 1;
+    p.n_reps        = 3;
+    p.exp           = 1;
+
+    int opt;
+    while((opt = getopt(argc, argv, "hw:e:x:m:n:o:p:")) >= 0) {
+        switch(opt) {
+        case 'h':
+        usage();
+        exit(0);
+        break;
+        case 'w': p.n_warmup      = atoi(optarg); break;
+        case 'e': p.n_reps        = atoi(optarg); break;
+        case 'x': p.exp           = atoi(optarg); break;
+        case 'm': p.m             = atoi(optarg); break;
+        case 'n': p.n             = atoi(optarg); break;
+        case 'o': p.M_            = atoi(optarg); break;
+        case 'p': p.N_            = atoi(optarg); break;
+        default:
+            fprintf(stderr, "\nUnrecognized option!\n");
+            usage();
+            exit(0);
+        }
+    }
+    assert(NR_DPUS > 0 && "Invalid # of dpus!");
+
+    return p;
+}
+#endif
diff --git a/TRNS/include/timer.h b/TRNS/include/timer.h
new file mode 100644
index 0000000..8d5c3d5
--- /dev/null
+++ b/TRNS/include/timer.h
@@ -0,0 +1,5 @@
+#pragma once
+
+#define N_TIMERS 10
+#include "../../include/timer_base.h"
+#undef N_TIMERS