From 5bc9dea242d68fa9942ef29a0e8511ad65c5f6a6 Mon Sep 17 00:00:00 2001
From: Robin Randhawa <robin.randhawa@arm.com>
Date: Fri, 15 Jan 2021 19:03:42 +0000
Subject: [PATCH] aarch64: context::switch: update the CONTEXT_SWITCH_LOCK

---
 src/context/arch/aarch64.rs | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/context/arch/aarch64.rs b/src/context/arch/aarch64.rs
index 5f5e2a3..ac74e22 100644
--- a/src/context/arch/aarch64.rs
+++ b/src/context/arch/aarch64.rs
@@ -239,6 +239,8 @@ impl Context {
 
         llvm_asm!("mov   $0, sp" : "=r"(self.sp) : : "memory" : "volatile");
         llvm_asm!("mov   sp, $0" : : "r"(next.sp) : "memory" : "volatile");
+
+        CONTEXT_SWITCH_LOCK.store(false, Ordering::SeqCst);
     }
 }