2 files changed, 12 insertions, 2 deletions
diff --git a/include/arch/cpu.h b/include/arch/cpu.h
index 8913b85..4fe0c99 100644
--- a/include/arch/cpu.h
+++ b/include/arch/cpu.h
@@ -8,6 +8,7 @@
  */
 
 #include <apos/types.h>
+#include <apos/tcb.h>
 
 #if defined(riscv64)
 #include "../../arch/riscv64/include/cpu.h"
@@ -24,6 +25,14 @@
  */
 id_t cpu_id();
 
-/** \todo add more cpu handling functions */
+/**
+ * Assign tcb to current cpu.
+ *
+ * @param t \ref tcb to assign to current cpu.
+ */
+void cpu_assign(struct tcb *t);
+
+/** \todo Should init be assigned one thread per core, or how should I handle
+ * giving out CPU IDs? */
 
 #endif /* APOS_CPU_H */
diff --git a/include/arch/timer.h b/include/arch/timer.h
index 71f878a..7f12100 100644
--- a/include/arch/timer.h
+++ b/include/arch/timer.h
@@ -18,9 +18,10 @@
 /**
  * Get hardware timer frequency.
  *
+ * @param fdt Global FDT pointer.
  * @return Hardware timer frequency, ticks/sec.
  */
-ticks_t stat_timer();
+ticks_t stat_timer(const void *fdt);
 
 /**
  * Set up timer interrupt for absolute ticks.