From a705a2e38bee035d6800999ba6eddd82792594f7 Mon Sep 17 00:00:00 2001
From: gingerBill <bill@gingerbill.org>
Date: Sun, 13 Nov 2022 22:55:32 +0000
Subject: Minor improvement to multi return value reducing stack usage

---
 src/check_stmt.cpp | 4 ++++
 1 file changed, 4 insertions(+)

(limited to 'src/check_stmt.cpp')

diff --git a/src/check_stmt.cpp b/src/check_stmt.cpp
index 9cacb4a35..8271ca451 100644
--- a/src/check_stmt.cpp
+++ b/src/check_stmt.cpp
@@ -2014,6 +2014,10 @@ void check_stmt_internal(CheckerContext *ctx, Ast *node, u32 flags) {
 		if (is_ast_decl(ds->stmt)) {
 			error(ds->token, "You cannot defer a declaration");
 		} else {
+			if (ctx->decl != nullptr) {
+				ctx->decl->defer_use_count += 1;
+			}
+
 			bool out_in_defer = ctx->in_defer;
 			ctx->in_defer = true;
 			check_stmt(ctx, ds->stmt, 0);
-- 
cgit v1.2.3