1 files changed, 4 insertions, 7 deletions
diff --git a/riscv/insns/sc_w.h b/riscv/insns/sc_w.h
index 4b4be50..fe4fcdc 100644
--- a/riscv/insns/sc_w.h
+++ b/riscv/insns/sc_w.h
@@ -1,10 +1,7 @@
 require_extension('A');
-if (MMU.check_load_reservation(RS1))
-{
-  MMU.store_uint32(RS1, RS2);
-  WRITE_RD(0);
-}
-else
-  WRITE_RD(1);
 
+bool have_reservation = MMU.check_load_reservation(RS1);
+MMU.amo_uint32(RS1, [&](uint32_t lhs) { return have_reservation ? RS2 : lhs; });
 MMU.yield_load_reservation();
+
+WRITE_RD(!have_reservation);