From 05001600966baeeba4ef42ddcf1804975fe88625 Mon Sep 17 00:00:00 2001 From: Birte Kristina Friesel Date: Tue, 13 May 2025 13:45:22 +0200 Subject: tracing: provide input_size and element_size in derived aspect --- VA/include/dfatool_host.ah | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) (limited to 'VA/include/dfatool_host.ah') diff --git a/VA/include/dfatool_host.ah b/VA/include/dfatool_host.ah index 6818333..7b33cec 100644 --- a/VA/include/dfatool_host.ah +++ b/VA/include/dfatool_host.ah @@ -4,12 +4,22 @@ #include "dfatool_host_dpu.ah" aspect DfatoolHostTiming : public DfatoolHostDPUTiming { + + DfatoolHostTiming() { + element_size = sizeof(T); + } + advice call("% input_params(...)") : after() { Params* p = tjp->result(); - printf("[==] VA | n_dpus=%u n_elements=%lu e_exp=%d \n", NR_DPUS, p->input_size, p->exp); + printf("[>>] VA | n_dpus=%u n_elements=%lu e_exp=%d\n", NR_DPUS, p->input_size, p->exp); + input_size = p->input_size; } advice call("% vector_addition_host(...)") : after() { - printf("[--] VA | n_dpus=%u n_ranks=%u n_elements=%lu n_elements_per_dpu=%lu e_exp=%d \n", n_dpus, n_ranks, p.input_size, p.input_size / n_dpus, p.exp); + printf("[--] VA | n_dpus=%u n_ranks=%u n_elements=%lu e_exp=%d\n", n_dpus, n_ranks, p.input_size, p.exp); + } + + advice execution("% main(...)") : after() { + printf("[<<] VA | n_dpus=%u n_elements=%lu e_exp=%d\n", NR_DPUS, p.input_size, p.exp); } }; -- cgit v1.2.3