1 files changed, 8 insertions, 11 deletions
diff --git a/test/monniaux/cycles.h b/test/monniaux/cycles.h
index 45c900e3..e4f098cd 100644
--- a/test/monniaux/cycles.h
+++ b/test/monniaux/cycles.h
@@ -1,28 +1,25 @@
 typedef unsigned long cycle_t;
 
 #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
+#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__
 #include <x86intrin.h>
 static inline cycle_t get_cycle(void) { return __rdtsc(); }