diff options
Diffstat (limited to 'test/monniaux/cycles.h')
-rw-r--r-- | test/monniaux/cycles.h | 87 |
1 files changed, 74 insertions, 13 deletions
diff --git a/test/monniaux/cycles.h b/test/monniaux/cycles.h index 45c900e3..c7dc582b 100644 --- a/test/monniaux/cycles.h +++ b/test/monniaux/cycles.h @@ -1,32 +1,93 @@ -typedef unsigned long cycle_t; +#include <stdint.h> +#include <inttypes.h> +#include <stdio.h> #ifdef __K1C__ -#ifdef __K1C_COS__ -#include <hal/cos_registers.h> -#define K1_SFR_PMC COS_SFR_PMC -#define K1_SFR_PM0 COS_SFR_PM0 -#else -#include <mppa_bare_runtime/k1c/registers.h> -#endif +typedef uint64_t cycle_t; +#define PRcycle PRId64 + +#include <../../k1-cos/include/hal/cos_registers.h> + static inline void cycle_count_config(void) { /* config pmc for cycle count */ - cycle_t pmc_value = __builtin_k1_get(K1_SFR_PMC); + cycle_t pmc_value = __builtin_k1_get(COS_SFR_PMC); pmc_value &= ~(0xfULL); - __builtin_k1_set(K1_SFR_PMC, pmc_value); + __builtin_k1_set(COS_SFR_PMC, pmc_value); } static inline cycle_t get_cycle(void) { - return __builtin_k1_get(K1_SFR_PM0); + return __builtin_k1_get(COS_SFR_PM0); } -#else + +#else // not K1C static inline void cycle_count_config(void) { } -#ifdef __x86_64__ + +#if defined(__i386__) || defined( __x86_64__) +#define PRcycle PRId64 +typedef uint64_t cycle_t; #include <x86intrin.h> static inline cycle_t get_cycle(void) { return __rdtsc(); } + +#elif __riscv +#ifdef __riscv32 +#define PRcycle PRId32 +typedef uint32_t cycle_t; +#else +#define PRcycle PRId64 +typedef uint64_t cycle_t; +#endif +static inline cycle_t get_cycle(void) { + cycle_t cycles; + asm volatile ("rdcycle %0" : "=r" (cycles)); + return cycles; +} + +#elif defined (__ARM_ARCH) && (__ARM_ARCH >= 6) +#if (__ARM_ARCH < 8) +typedef uint32_t cycle_t; +#define PRcycle PRId32 + +/* need this kernel module +https://github.com/zertyz/MTL/tree/master/cpp/time/kernel/arm */ +static inline cycle_t get_cycle(void) { + cycle_t cycles; + __asm__ volatile ("mrc p15, 0, %0, c9, c13, 0":"=r" (cycles)); + return cycles; +} #else +#define PRcycle PRId64 +typedef uint64_t cycle_t; +/* need this kernel module: +https://github.com/jerinjacobk/armv8_pmu_cycle_counter_el0 + +on 5+ kernels, remove first argument of access_ok macro */ + +static inline cycle_t get_cycle(void) +{ + uint64_t val; + __asm__ volatile("mrs %0, pmccntr_el0" : "=r"(val)); + return val; +} +#endif + +#else +#define PRcycle PRId32 +typedef uint32_t cycle_t; static inline cycle_t get_cycle(void) { return 0; } #endif #endif + +#ifdef MAX_MEASURES + #define TIMEINIT(i) {_last_stop[i] = get_cycle();} + #define TIMESTOP(i) {cycle_t cur = get_cycle(); _total_cycles[i] += cur - _last_stop[i]; _last_stop[i] = cur;} + #define TIMEPRINT(n) { for (int i = 0; i <= n; i++) printf("%d cycles: %" PRIu64 "\n", i, _total_cycles[i]); } +#endif + + +#ifdef MAX_MEASURES + static cycle_t _last_stop[MAX_MEASURES] = {0}; + static cycle_t _total_cycles[MAX_MEASURES] = {0}; +#endif |