avr: support sleep

1 files changed, 11 insertions, 2 deletions

diff --git a/embassy-executor/src/arch/avr.rs b/embassy-executor/src/arch/avr.rs
index 2c715f297..fa6afe762 100644
--- a/embassy-executor/src/arch/avr.rs
+++ b/embassy-executor/src/arch/avr.rs
@@ -8,11 +8,16 @@ mod thread {
     use core::marker::PhantomData;
 
     pub use embassy_executor_macros::main_avr as main;
+    use portable_atomic::{AtomicBool, Ordering};
 
     use crate::{raw, Spawner};
 
+    static SIGNAL_WORK_THREAD_MODE: AtomicBool = AtomicBool::new(false);
+
     #[export_name = "__pender"]
-    fn __pender(_context: *mut ()) {}
+    fn __pender(_context: *mut ()) {
+        SIGNAL_WORK_THREAD_MODE.store(true, Ordering::SeqCst);
+    }
 
     /// avr Executor
     pub struct Executor {
@@ -52,7 +57,11 @@ mod thread {
 
             loop {
                 unsafe {
-                    self.inner.poll();
+                    if SIGNAL_WORK_THREAD_MODE.swap(false, Ordering::SeqCst) {
+                        self.inner.poll();
+                    } else {
+                        avr_device::asm::sleep();
+                    }
                 }
             }
         }