BS: Add AspectC++ support

4 files changed, 145 insertions, 0 deletions

diff --git a/BS/include/common.h b/BS/include/common.h
new file mode 100644
index 0000000..d0b2865
--- /dev/null
+++ b/BS/include/common.h
@@ -0,0 +1,54 @@
+#ifndef _COMMON_H_
+#define _COMMON_H_
+
+#ifdef TL
+#define TASKLETS_INITIALIZER TASKLETS(TL, main, 2048, 2)
+#define NB_OF_TASKLETS_PER_DPU TL
+#else
+#define TASKLETS_INITIALIZER TASKLETS(16, main, 2048, 2)
+#define NB_OF_TASKLETS_PER_DPU 16
+#endif
+
+// Transfer size between MRAM and WRAM
+// BL=10 does not work here.
+#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)
+#endif
+
+// Data type
+#define DTYPE int64_t
+
+// Vector size
+#ifndef INPUT_SIZE
+#define INPUT_SIZE 2048576
+#endif
+
+enum kernel {
+	kernel1 = 0,
+	nr_kernels = 1,
+};
+
+typedef struct {
+	uint64_t input_size;
+	uint64_t slice_per_dpu;
+	enum kernel kernel;
+} dpu_arguments_t;
+
+// Structures used by both the host and the dpu to communicate information
+typedef struct {
+	DTYPE found;
+} dpu_results_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"
+#endif
diff --git a/BS/include/dfatool_host.ah b/BS/include/dfatool_host.ah
new file mode 100644
index 0000000..2f55c40
--- /dev/null
+++ b/BS/include/dfatool_host.ah
@@ -0,0 +1,27 @@
+#pragma once
+
+#include <sys/time.h>
+#include "dfatool_host_dpu.ah"
+
+aspect DfatoolHostTiming : public DfatoolHostDPUTiming {
+
+	virtual int getKernel() { return 1; }
+
+	DfatoolHostTiming() {
+		element_size = sizeof(uint32_t);
+	}
+
+	advice call("% input_params(...)"): after() {
+		Params* p = tjp->result();
+		input_size = INPUT_SIZE + p->num_querys;
+		printf("[>>] BS | n_dpus=%u n_elements=%lu\n", NR_DPUS, input_size);
+	}
+
+	advice call("% binarySearch(...)") : after() {
+		printf("[--] BS | n_dpus=%u n_elements=%lu\n", NR_DPUS, input_size);
+	}
+
+	advice execution("% main(...)") : after() {
+		printf("[<<] BS | n_dpus=%u n_elements=%lu\n", NR_DPUS, input_size);
+	}
+};
diff --git a/BS/include/params.h b/BS/include/params.h
new file mode 100644
index 0000000..f970eda
--- /dev/null
+++ b/BS/include/params.h
@@ -0,0 +1,59 @@
+#ifndef _PARAMS_H_
+#define _PARAMS_H_
+
+#include "common.h"
+
+typedef struct Params {
+	long num_querys;
+	unsigned n_warmup;
+	unsigned n_reps;
+} Params;
+
+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"
+		"\nBenchmark-specific options:"
+		"\n    -i <I>    problem size (default=%d queries)" "\n", PROBLEM_SIZE);
+}
+
+struct Params input_params(int argc, char **argv)
+{
+	struct Params p;
+	p.num_querys = PROBLEM_SIZE;
+	p.n_warmup = 1;
+	p.n_reps = 3;
+
+	int opt;
+	while ((opt = getopt(argc, argv, "h:i:w:e:")) >= 0) {
+		switch (opt) {
+		case 'h':
+			usage();
+			exit(0);
+			break;
+		case 'i':
+			p.num_querys = atol(optarg);
+			break;
+		case 'w':
+			p.n_warmup = atoi(optarg);
+			break;
+		case 'e':
+			p.n_reps = 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/BS/include/timer.h b/BS/include/timer.h
new file mode 100644
index 0000000..7b80823
--- /dev/null
+++ b/BS/include/timer.h
@@ -0,0 +1,5 @@
+#pragma once
+
+#define N_TIMERS 7
+#include "../../include/timer_base.h"
+#undef N_TIMERS