Add support for cacheless PIM

pim-os/src/pim/state.rs

@@ -1,10 +1,11 @@
+use crate::boot;
+
 use super::config::PimWriter;
 use pim_isa::{BankMode, Kernel, PimConfig};
 
 #[derive(Debug)]
 pub struct PimState {
     kernel: Kernel,
-    bank_mode: BankMode,
     writer: PimWriter,
 }
 
@@ -12,18 +13,24 @@ impl PimState {
     pub fn new(kernel: &Kernel) -> Self {
         Self {
             kernel: kernel.clone(),
-            bank_mode: BankMode::SingleBank,
             writer: PimWriter,
         }
     }
 
     // TODO return token and return to singlebank when dropped
     pub fn set_bank_mode(&mut self, bank_mode: BankMode) {
-        self.bank_mode = bank_mode;
+        if cfg!(cacheless) {
+            match bank_mode {
+                BankMode::SingleBank => unsafe { boot::set_page_table_cache() },
+                BankMode::AllBank => (),
+                BankMode::PimAllBank => unsafe { boot::set_page_table_non_cache() },
+            }
+        }
+
         self.writer.write(
             serde_json_core::to_string::<PimConfig, 1024>(&PimConfig {
                 kernel: self.kernel.clone(),
-                bank_mode: self.bank_mode,
+                bank_mode,
             })
             .unwrap()
             .as_str(),