TRNS: refactor AspectC++ and timer headers

1 files changed, 25 insertions, 0 deletions

diff --git a/TRNS/include/dfatool_host.ah b/TRNS/include/dfatool_host.ah
new file mode 100644
index 0000000..94f6101
--- /dev/null
+++ b/TRNS/include/dfatool_host.ah
@@ -0,0 +1,25 @@
+#pragma once
+
+#include <sys/time.h>
+#include "dfatool_host_dpu.ah"
+
+aspect DfatoolHostTiming : public DfatoolHostDPUTiming {
+
+	DfatoolHostTiming() {
+		element_size = sizeof(T);
+	}
+
+	advice call("% input_params(...)") : after() {
+		Params* p = tjp->result();
+		input_size = p->M_ * p->m * p->N_ * p->n;
+		printf("[>>] TRNS | n_dpus=%u n_elements=%lu e_exp=%d\n", NR_DPUS, input_size, p->exp);
+	}
+
+	advice call("% trns_host(...)") : after() {
+		printf("[--] TRNS | n_dpus=%u n_ranks=%u n_elements=%lu e_exp=%d\n", n_dpus, n_ranks, input_size, p.exp);
+	}
+
+	advice execution("% main(...)") : after() {
+		printf("[<<] TRNS | n_dpus=%u n_elements=%lu e_exp=%d\n", NR_DPUS, input_size, p.exp);
+	}
+};