TS: Add AspectC++ support

1 files changed, 16 insertions, 0 deletions

diff --git a/include/dfatool_host_dpu.ah b/include/dfatool_host_dpu.ah
index 560e410..0e03d31 100644
--- a/include/dfatool_host_dpu.ah
+++ b/include/dfatool_host_dpu.ah
@@ -88,6 +88,22 @@ aspect DfatoolHostDPUTiming {
 		);
 	}
 
+	advice call("% dpu_copy_to(...)") : around() {
+		size_t payload_size = *(tjp->arg<4>());
+		gettimeofday(&starttime, NULL);
+		tjp->proceed();
+		gettimeofday(&stoptime, NULL);
+		double time_us = (stoptime.tv_sec - starttime.tv_sec) * 1000000.0 + (stoptime.tv_usec - starttime.tv_usec);
+		printf("[::] dpu_copy_to @ %s:%d | n_dpus=%u n_ranks=%u payload_B=%lu | latency_us=%f throughput_MiBps=%f\n",
+			tjp->filename(),
+			tjp->line(),
+			n_dpus, n_ranks,
+			payload_size,
+			time_us,
+			payload_size / (time_us * M_to_Mi)
+		);
+	}
+
 	advice call("% dpu_push_xfer(...)") : around() {
 		size_t payload_size = *(tjp->arg<4>());
 		gettimeofday(&starttime, NULL);