#include <inttypes.h>
#include <sys/mman.h>
#include <stdbool.h>
+#include <pthread.h>
+#include <stdlib.h>
#ifndef SYS_prem_memguard_check
#define SYS_prem_memguard_check 793
#define MGRET_OVER_MEM_MASK 0x1lu
#define MGRET_OVER_TIM_MASK 0x1lu
+#define MAX_CORES 6
+
char memory[128*10204*1024];
+static pthread_barrier_t barrier;
+
struct mg_ret {
uint32_t time;
uint32_t mem;
flags & MGF_MASK_INT ? 'I' : '-'
);
struct mg_ret r = mgret(retval);
- printf("%-40s %-25s ⇒ time:%8u%c mem:%8u%c\n",
+ printf("Testing \"%-40s %-25s ⇒ time:%8u%c mem:%8u%c\" in %s:\n",
call, code,
r.time, r.time_ovf ? '!' : ' ',
- r.mem, r.mem_ovf ? '!' : ' ');
+ r.mem, r.mem_ovf ? '!' : ' ',
+ __FILE__);
+}
+
+void wvtest_pass(bool cond, const char* file, int line, const char* str)
+{
+ printf("! %s:%d %s %s\n", file, line, str, cond ? "ok" : "FAILURE");
}
+#define WVPASS(cond) wvtest_pass(cond, __FILE__, __LINE__, #cond)
+
#define MGTEST(timeout_us, mem_budget, flags, code) \
({ \
long retval; \
mgret(retval); \
})
-int main(int argc, char *argv[])
+void *test_thread(void *ptr)
{
- cpu_set_t set;
- int cpu = 0;
-
- for (int i = 0; i < sizeof(memory); i += 64)
- memory[i] = 1;
-
- if (argc > 1)
- cpu = atoi(argv[1]);
-
+ cpu_set_t set;
+ int *cpu;
+ cpu = (int *) ptr;
+
/* Ensure that memory phase starts and ends on the same CPU */
CPU_ZERO(&set);
- CPU_SET(cpu, &set);
+ CPU_SET(*cpu, &set);
if (sched_setaffinity(getpid(), sizeof(set), &set) < 0)
err(1, "sched_setaffinity");
- printf("Pinned to CPU %d\n", cpu);
-
+ printf("Pinned to CPU %d\n", *cpu);
+
+ pthread_barrier_wait(&barrier);
+
+ struct mg_ret r;
for (uint64_t flags = 0; flags < 4; flags++) {
compute_kernel(1); /* warm up */
- MGTEST(5000, 10000, flags, compute_kernel(1000));
- MGTEST(500, 10000, flags, compute_kernel(1000));
- MGTEST(5000, 10000, flags, compute_kernel(2000));
- MGTEST(500, 10000, flags, compute_kernel(2000));
-
- MGTEST(100000, 500000, flags, read_memory(100000));
- MGTEST(100000, 50000, flags, read_memory(100000));
- MGTEST(100000, 500000, flags, read_memory_rnd(100000));
- MGTEST(100000, 50000, flags, read_memory_rnd(100000));
- MGTEST(100000, 500000, flags, write_memory(100000));
- MGTEST(100000, 50000, flags, write_memory(100000));
-
- MGTEST(100000, 5000000, flags, read_memory(1000000));
- MGTEST(100000, 500000, flags, read_memory(1000000));
- MGTEST(100000, 5000000, flags, read_memory_rnd(1000000));
- MGTEST(100000, 500000, flags, read_memory_rnd(1000000));
- MGTEST(100000, 5000000, flags, write_memory(1000000));
- MGTEST(100000, 500000, flags, write_memory(1000000));
+
+ ///////////////////////////////////////////////////////
+ r = MGTEST(5000, 10000, flags, compute_kernel(1000));
+ WVPASS(!r.time_ovf);
+ WVPASS(r.time > 900);
+
+ r = MGTEST(500, 10000, flags, compute_kernel(1000));
+ WVPASS(r.time_ovf);
+ WVPASS(r.time > 900);
+
+ r = MGTEST(5000, 10000, flags, compute_kernel(2000));
+ WVPASS(!r.time_ovf);
+ WVPASS(r.time > 1900);
+
+ r = MGTEST(500, 10000, flags, compute_kernel(2000));
+ WVPASS(r.time_ovf);
+ WVPASS(r.time > 1900);
+
+ ///////////////////////////////////////////////////////
+ r = MGTEST(100000, 500000, flags, read_memory(100000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+ r = MGTEST(100000, 50000, flags, read_memory(100000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+ r = MGTEST(100000, 500000, flags, read_memory_rnd(100000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+ r = MGTEST(100000, 50000, flags, read_memory_rnd(100000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+ r = MGTEST(100000, 500000, flags, write_memory(100000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+ r = MGTEST(100000, 50000, flags, write_memory(100000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 100000);
+
+
+ ///////////////////////////////////////////////////////
+ r = MGTEST(100000, 5000000, flags, read_memory(1000000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 900000);
+
+ r = MGTEST(100000, 500000, flags, read_memory(1000000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 900000);
+
+ r = MGTEST(100000, 5000000, flags, read_memory_rnd(1000000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 900000);
+
+ r = MGTEST(100000, 500000, flags, read_memory_rnd(1000000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 900000);
+
+ r = MGTEST(100000, 5000000, flags, write_memory(1000000));
+ WVPASS(!r.mem_ovf);
+ WVPASS(r.mem >= 900000);
+
+ r = MGTEST(100000, 500000, flags, write_memory(1000000));
+ WVPASS(r.mem_ovf);
+ WVPASS(r.mem >= 900000);
printf("\n");
}
+
+ return NULL;
+}
+
+int main(int argc, char *argv[])
+{
+ int cpu_mask = 0;
+
+ int cpu_count = 0;
+ pthread_t threads[MAX_CORES];
+ int cores[MAX_CORES];
+ int retvals[MAX_CORES];
+
+ /* TODO: currently shared memory */
+ for (int i = 0; i < sizeof(memory); i += 64)
+ memory[i] = 1;
+
+ if (argc > 1)
+ cpu_mask = strtol(argv[1], NULL, 16);
+
+ for(int i=0; i<MAX_CORES; i++){
+ if(cpu_mask & (1 << i)){
+ cpu_count++;
+ }
+ }
+
+ int s = pthread_barrier_init(&barrier, NULL, cpu_count);
+ if (s != 0)
+ err(1, "pthread_barrier_init");
+
+ for(int i=0; i<MAX_CORES; i++){
+ if(cpu_mask & (1 << i)){
+ cores[i] = i;
+ retvals[i] = pthread_create(&threads[i], NULL, test_thread, (void*) &cores[i]);
+ }
+ }
+
+ for(int i=0; i<MAX_CORES; i++){
+ if(cpu_mask & (1 << i)){
+ pthread_join(threads[i], NULL);
+ }
+ }
+
return 0;
}