avr: sleep fix

1 files changed, 5 insertions, 2 deletions

diff --git a/embassy-executor/src/arch/avr.rs b/embassy-executor/src/arch/avr.rs
index fa6afe762..11e81ed9a 100644
--- a/embassy-executor/src/arch/avr.rs
+++ b/embassy-executor/src/arch/avr.rs
@@ -57,10 +57,13 @@ mod thread {
 
             loop {
                 unsafe {
+                    avr_device::interrupt::disable();
                     if SIGNAL_WORK_THREAD_MODE.swap(false, Ordering::SeqCst) {
-                        self.inner.poll();
-                    } else {
+                        avr_device::interrupt::enable();
                         avr_device::asm::sleep();
+                    } else {
+                        avr_device::interrupt::enable();
+                        self.inner.poll();
                     }
                 }
             }