blob: 6ea4a1860b3139f5afa92dc7ae980c65b72e6d1c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
#pragma once
#include <sys/time.h>
#include "dfatool_host_dpu.ah"
aspect DfatoolHostTiming : public DfatoolHostDPUTiming {
unsigned int n_rows, n_cols;
unsigned int element_size;
virtual int getKernel() { return 1; }
DfatoolHostTiming() {
element_size = sizeof(uint32_t);
}
advice call("% input_params(...)"): after() {
Params* p = tjp->result();
n_rows = p->m_size;
n_cols = p->n_size;
printf("[>>] MLP | n_dpus=%u n_rows=%u n_cols=%u\n", NR_DPUS, n_rows, n_cols);
}
advice call("% start(...)") : after() {
if (*(tjp->arg<1>()) == 1) {
printf("[--] MLP | n_dpus=%u n_rows=%u n_cols=%u\n", NR_DPUS, n_rows, n_cols);
}
}
advice execution("% main(...)") : after() {
printf("[<<] MLP | n_dpus=%u n_rows=%u n_cols=%u\n", NR_DPUS, n_rows, n_cols);
}
};
|