cleanup

arch/x86_64/sched.c

@@ -13,6 +13,7 @@
 #include <x86_64/defs.h>
 #include <x86_64/gdt.h>
 #include <x86_64/idt.h>
+#include <x86_64/percpu.h>
 
 // The task frame consists of registers that the C ABI marks as callee-saved.
 // If we don't save them, these registers are lost during a context switch.
@@ -53,7 +54,7 @@ static struct task* perform_switch(uint64_t* old_rsp, uint64_t new_rsp, struct t
 }
 
 struct task* arch_sched_switch(struct task* from, struct task* to) {
-    struct arch_percpu* cpu = &percpu_get()->arch;
+    struct arch_percpu* cpu = percpu_ref(arch_data);
     struct arch_task_context* from_ctx = &from->context;
     struct arch_task_context* to_ctx = &to->context;
 
@@ -108,7 +109,7 @@ errno_t arch_task_init(
     uintptr_t stack_start,
     bool is_user
 ) {
-    struct arch_percpu* cpu = &percpu_get()->arch;
+    struct arch_percpu* cpu = percpu_ref(arch_data);
 
     // Prepare a dummy stack with an entry point function to return to.
     struct task_frame* frame = (struct task_frame*)stack_start - 1;