From 89b6b09be574021e58fbebfa92cbc9c390f85acc Mon Sep 17 00:00:00 2001
From: lisyarus <lisyarus@gmail.com>
Date: Fri, 13 Mar 2026 14:25:05 +0300
Subject: [PATCH] Fix scope stack offset cleanup in aarch64 compiler

---
 libs/jit/source/arch/aarch64/compiler.cpp | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libs/jit/source/arch/aarch64/compiler.cpp b/libs/jit/source/arch/aarch64/compiler.cpp
index e3d6206..313c23e 100644
--- a/libs/jit/source/arch/aarch64/compiler.cpp
+++ b/libs/jit/source/arch/aarch64/compiler.cpp
@@ -935,7 +935,10 @@ namespace pslang::jit::aarch64
 			void scope_cleanup()
 			{
 				if (scopes.back().stack_offset > 0)
+				{
 					builder.add_imm(31, 31, scopes.back().stack_offset);
+					stack_offset -= scopes.back().stack_offset;
+				}
 			}
 		};