@scriptc/compiler 0.2.4 → 0.2.6

package/dist/backend/llvm/stack-callbacks.d.ts

@@ -0,0 +1,14 @@
+import type { IrExpr, IrFunction } from "../../ir/ir.js";
+import type { LlValue, LlvmEmitterContext } from "./expr-context.js";
+/** A synchronous, invocation-only consumer cannot retain the environment.
+ * Capture boxes still have independent owners in the ordinary call frame;
+ * only the closure allocation and its cycle header disappear. */
+export declare class StackCallbacks {
+    private readonly functions;
+    private readonly targets;
+    private remaining;
+    constructor(functions: ReadonlyMap<string, IrFunction>);
+    reset(enabled: boolean): void;
+    emit(host: LlvmEmitterContext, value: IrExpr): LlValue | null;
+    eligible(name: string): boolean;
+}

package/dist/backend/llvm/stack-callbacks.js

@@ -0,0 +1,73 @@
+import { everyStmtList } from "../../ir/traverse.js";
+import { mangleLocal } from "../mangle.js";
+/** A synchronous, invocation-only consumer cannot retain the environment.
+ * Capture boxes still have independent owners in the ordinary call frame;
+ * only the closure allocation and its cycle header disappear. */
+export class StackCallbacks {
+    functions;
+    targets = new Map();
+    remaining = 0;
+    constructor(functions) {
+        this.functions = functions;
+    }
+    reset(enabled) {
+        // Bound entry-block storage even when a function contains many sites.
+        this.remaining = enabled ? 4096 : 0;
+    }
+    emit(host, value) {
+        if (value.kind !== "closure" || value.captures.length > 32)
+            return null;
+        const bytes = (5 + value.captures.length) * (host.sizeType === "i64" ? 8 : 4);
+        if (bytes > this.remaining || !this.eligible(value.fnName))
+            return null;
+        this.remaining -= bytes;
+        const B = host.B;
+        const closure = B.slot();
+        B.entryAllocas.push(`${closure} = alloca { %ScrClosure, [${value.captures.length} x ptr] }`);
+        B.line(`store %ScrClosure zeroinitializer, ptr ${closure}`);
+        B.line(`store ${host.sizeType} -1, ptr ${closure}`);
+        const fn = B.tmp();
+        B.line(`${fn} = getelementptr inbounds %ScrClosure, ptr ${closure}, i32 0, i32 1`);
+        B.line(`store ptr @${host.callTarget(value.fnName)}, ptr ${fn}`);
+        const count = B.tmp();
+        B.line(`${count} = getelementptr inbounds %ScrClosure, ptr ${closure}, i32 0, i32 2`);
+        B.line(`store ${host.sizeType} ${value.captures.length}, ptr ${count}`);
+        if (this.functions.get(value.fnName)?.ownsPrototype) {
+            const kind = B.tmp();
+            B.line(`${kind} = getelementptr inbounds %ScrClosure, ptr ${closure}, i32 0, i32 4`);
+            B.line(`store i32 4, ptr ${kind}`);
+        }
+        value.captures.forEach((localId, index) => {
+            const box = host.retainBox(host.loadBox(`%${mangleLocal(localId)}`));
+            const caps = B.tmp();
+            const slot = B.tmp();
+            B.line(`${caps} = getelementptr inbounds %ScrClosure, ptr ${closure}, i32 1`);
+            B.line(`${slot} = getelementptr inbounds ptr, ptr ${caps}, ${host.sizeType} ${index}`);
+            B.line(`store ptr ${box}, ptr ${slot}`);
+            // Use the same forward capture order as scr_closure_release, on both
+            // normal and exceptional exits. The immortal stack header owns none.
+            host.own({ name: slot, type: value.type, slot: true, boxed: true });
+        });
+        return { name: closure, type: value.type };
+    }
+    eligible(name) {
+        const cached = this.targets.get(name);
+        if (cached !== undefined)
+            return cached;
+        const fn = this.functions.get(name);
+        let nodes = 0;
+        // A self-reference can publish the running environment even when its
+        // caller only invokes it. Async/generator bodies can outlive the call.
+        const safe = fn !== undefined &&
+            fn.captures !== undefined &&
+            !fn.async &&
+            !fn.generator &&
+            everyStmtList(fn.body, {
+                expr: (expr) => ++nodes <= 512 && expr.kind !== "selfRef",
+                stmt: () => ++nodes <= 512,
+            });
+        this.targets.set(name, safe);
+        return safe;
+    }
+}
+//# sourceMappingURL=stack-callbacks.js.map

package/dist/backend/llvm/stack-callbacks.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stack-callbacks.js","sourceRoot":"","sources":["../../../src/backend/llvm/stack-callbacks.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AACrD,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAG3C;;iEAEiE;AACjE,MAAM,OAAO,cAAc;IAII,SAAS;IAHrB,OAAO,GAAG,IAAI,GAAG,EAAmB,CAAC;IAC9C,SAAS,GAAG,CAAC,CAAC;IAEtB,YAA6B,SAA0C;yBAA1C,SAAS;IAAoC,CAAC;IAE3E,KAAK,CAAC,OAAgB;QACpB,sEAAsE;QACtE,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACtC,CAAC;IAED,IAAI,CAAC,IAAwB,EAAE,KAAa;QAC1C,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,EAAE;YAAE,OAAO,IAAI,CAAC;QACxE,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9E,IAAI,KAAK,GAAG,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,CAAC;YAAE,OAAO,IAAI,CAAC;QACxE,IAAI,CAAC,SAAS,IAAI,KAAK,CAAC;QACxB,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACjB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,OAAO,6BAA6B,KAAK,CAAC,QAAQ,CAAC,MAAM,WAAW,CAAC,CAAC;QAC7F,CAAC,CAAC,IAAI,CAAC,0CAA0C,OAAO,EAAE,CAAC,CAAC;QAC5D,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,YAAY,OAAO,EAAE,CAAC,CAAC;QACpD,MAAM,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;QACnB,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,8CAA8C,OAAO,gBAAgB,CAAC,CAAC;QACnF,CAAC,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;QACjE,MAAM,KAAK,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;QACtB,CAAC,CAAC,IAAI,CAAC,GAAG,KAAK,8CAA8C,OAAO,gBAAgB,CAAC,CAAC;QACtF,CAAC,CAAC,IAAI,CAAC,SAAS,IAAI,CAAC,QAAQ,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,SAAS,KAAK,EAAE,CAAC,CAAC;QACxE,IAAI,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,aAAa,EAAE,CAAC;YACpD,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,8CAA8C,OAAO,gBAAgB,CAAC,CAAC;YACrF,CAAC,CAAC,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;QACrC,CAAC;QACD,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE;YACxC,MAAM,GAAG,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,WAAW,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;YACrE,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACrB,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,8CAA8C,OAAO,SAAS,CAAC,CAAC;YAC9E,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,sCAAsC,IAAI,KAAK,IAAI,CAAC,QAAQ,IAAI,KAAK,EAAE,CAAC,CAAC;YACvF,CAAC,CAAC,IAAI,CAAC,aAAa,GAAG,SAAS,IAAI,EAAE,CAAC,CAAC;YACxC,qEAAqE;YACrE,qEAAqE;YACrE,IAAI,CAAC,GAAG,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QACH,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7C,CAAC;IAED,QAAQ,CAAC,IAAY;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,MAAM,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC;QACxC,MAAM,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACpC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,qEAAqE;QACrE,uEAAuE;QACvE,MAAM,IAAI,GACR,EAAE,KAAK,SAAS;YAChB,EAAE,CAAC,QAAQ,KAAK,SAAS;YACzB,CAAC,EAAE,CAAC,KAAK;YACT,CAAC,EAAE,CAAC,SAAS;YACb,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;gBACrB,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,KAAK,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS;gBACzD,IAAI,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,IAAI,GAAG;aAC3B,CAAC,CAAC;QACL,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;QAC7B,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}

package/dist/backend/llvm/stack-captures.d.ts

@@ -0,0 +1,25 @@
+import { type IrFunction, type IrType } from "../../ir/ir.js";
+import type { CallLifetimes } from "./call-lifetimes.js";
+import type { LlvmEmitterContext } from "./expr-context.js";
+import type { StackCallbacks } from "./stack-callbacks.js";
+/** A captured binding can stay in its declaring frame only when every
+ * environment that contains it finishes within that binding's scope.
+ * Payload values can still escape: their ordinary owned reads retain them. */
+export declare class StackCaptures {
+    private readonly functions;
+    private readonly lifetimes;
+    private readonly callbacks;
+    private readonly recaptured;
+    private locals;
+    private remaining;
+    constructor(functions: ReadonlyMap<string, IrFunction>, lifetimes: CallLifetimes, callbacks: StackCallbacks);
+    reset(fn: IrFunction, enabled: boolean): void;
+    emit(host: LlvmEmitterContext, id: string, type: IrType): {
+        box: string;
+        owner?: {
+            slot: string;
+            type: IrType;
+        };
+    } | null;
+    private nestedCaptures;
+}

package/dist/backend/llvm/stack-captures.js

@@ -0,0 +1,150 @@
+import { isRefCounted, } from "../../ir/ir.js";
+import { everyExprChild, everyStmtChild, everyStmtList } from "../../ir/traverse.js";
+import { vAdapters } from "./shapes.js";
+/** A captured binding can stay in its declaring frame only when every
+ * environment that contains it finishes within that binding's scope.
+ * Payload values can still escape: their ordinary owned reads retain them. */
+export class StackCaptures {
+    functions;
+    lifetimes;
+    callbacks;
+    recaptured = new Map();
+    locals = new Set();
+    remaining = 0;
+    constructor(functions, lifetimes, callbacks) {
+        this.functions = functions;
+        this.lifetimes = lifetimes;
+        this.callbacks = callbacks;
+    }
+    reset(fn, enabled) {
+        this.locals = new Set();
+        this.remaining = enabled ? 2048 : 0;
+        if (!enabled)
+            return;
+        const possible = new Set(fn.locals.filter((local) => local.boxed && !local.tdz).map((local) => local.id));
+        if (possible.size === 0)
+            return;
+        const invalid = new Set([
+            ...fn.params.map((param) => param.localId),
+            ...(fn.captures ?? []).map((capture) => capture.localId),
+            ...(fn.classCaptures ?? []).map((capture) => capture.localId),
+        ]);
+        const declarations = new Map();
+        let nodes = 0;
+        const expr = (node, immediate) => {
+            if (++nodes > 4096)
+                return false;
+            if (node.kind === "closure") {
+                const target = this.functions.get(node.fnName);
+                const safe = immediate && this.callbacks.eligible(node.fnName);
+                const nested = safe ? this.nestedCaptures(target) : null;
+                node.captures.forEach((id, index) => {
+                    const capture = target?.captures?.[index];
+                    if (!safe || !capture || nested.has(capture.localId))
+                        invalid.add(id);
+                });
+            }
+            else if (node.kind === "classRef") {
+                for (const id of node.captures ?? [])
+                    invalid.add(id);
+            }
+            else if (node.kind === "call") {
+                const parameters = this.lifetimes.parameters.get(node.callee);
+                return node.args.every((arg, index) => expr(arg, parameters?.has(index) === true));
+            }
+            else if (node.kind === "callValue") {
+                return (expr(node.callee, true) &&
+                    (!node.receiver || expr(node.receiver, false)) &&
+                    node.args.every((arg) => expr(arg, false)));
+            }
+            return everyExprChild(node, (child) => expr(child, false), stmt);
+        };
+        const stmt = (node) => {
+            if (++nodes > 4096)
+                return false;
+            if (node.kind === "varDecl")
+                declarations.set(node.localId, (declarations.get(node.localId) ?? 0) + 1);
+            else if (node.kind === "forOf")
+                invalid.add(node.localId);
+            else if (node.kind === "tryCatch" && node.catchLocalId !== null)
+                invalid.add(node.catchLocalId);
+            else if (node.kind === "switch") {
+                // Dispatch can skip a declaration on a later loop iteration. The
+                // shared switch scope resets binding slots, not stack payload slots.
+                for (const branch of node.cases) {
+                    for (const child of branch.body) {
+                        if (child.kind === "varDecl")
+                            invalid.add(child.localId);
+                    }
+                }
+            }
+            else if (node.kind === "for") {
+                // Loop-head lets are freshened by a separate allocation path.
+                const initializers = node.init?.kind === "block" ? node.init.body : node.init ? [node.init] : [];
+                for (const init of initializers)
+                    if (init.kind === "varDecl")
+                        invalid.add(init.localId);
+            }
+            return everyStmtChild(node, (child) => expr(child, false), stmt);
+        };
+        if (!fn.body.every(stmt))
+            return;
+        for (const id of possible) {
+            if (!invalid.has(id) && declarations.get(id) === 1)
+                this.locals.add(id);
+        }
+    }
+    emit(host, id, type) {
+        // 48 bytes also bounds the smaller wasm32 layout. Charge actual emitted
+        // sites, including duplicated finally bodies and versioned loops.
+        if (!this.locals.has(id) || this.remaining < 48)
+            return null;
+        const ref = isRefCounted(type);
+        if (!ref &&
+            type.kind !== "f64" &&
+            type.kind !== "date" &&
+            type.kind !== "procStream" &&
+            type.kind !== "bool")
+            return null;
+        this.remaining -= 48;
+        const B = host.B;
+        const box = B.slot();
+        B.entryAllocas.push(`${box} = alloca %ScrBox`);
+        B.entryAllocas.push(`store %ScrBox zeroinitializer, ptr ${box}`);
+        let retain = "null", release = "null";
+        if (ref) {
+            const adapters = vAdapters(host.shapeHost, type);
+            retain = adapters.retain;
+            release = adapters.release;
+        }
+        const kind = ref ? 5 : type.kind === "bool" ? 1 : 0;
+        B.line(`store %ScrBox { ${host.sizeType} -1, i32 ${kind}, ptr ${retain}, ptr ${release}, ptr null, i64 0 }, ptr ${box}`);
+        if (!ref)
+            return { box };
+        const payload = B.slot();
+        B.entryAllocas.push(`${payload} = getelementptr inbounds %ScrBox, ptr ${box}, i32 0, i32 5`);
+        // The immortal box is invisible to cycle tracing. Its one payload owner
+        // is an ordinary external root, released at the original scope boundary.
+        return { box, owner: { slot: payload, type } };
+    }
+    nestedCaptures(fn) {
+        const cached = this.recaptured.get(fn.name);
+        if (cached)
+            return cached;
+        const captures = new Set();
+        // eligible() already bounded this target's complete body traversal.
+        everyStmtList(fn.body, {
+            stmt: () => true,
+            expr: (node) => {
+                if (node.kind === "closure" || node.kind === "classRef") {
+                    for (const id of node.captures ?? [])
+                        captures.add(id);
+                }
+                return true;
+            },
+        });
+        this.recaptured.set(fn.name, captures);
+        return captures;
+    }
+}
+//# sourceMappingURL=stack-captures.js.map

package/dist/backend/llvm/stack-captures.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stack-captures.js","sourceRoot":"","sources":["../../../src/backend/llvm/stack-captures.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,GAKb,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,cAAc,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAGrF,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAGxC;;8EAE8E;AAC9E,MAAM,OAAO,aAAa;IAML,SAAS;IACT,SAAS;IACT,SAAS;IAPX,UAAU,GAAG,IAAI,GAAG,EAAuB,CAAC;IACrD,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IAC3B,SAAS,GAAG,CAAC,CAAC;IAEtB,YACmB,SAA0C,EAC1C,SAAwB,EACxB,SAAyB;yBAFzB,SAAS;yBACT,SAAS;yBACT,SAAS;IACzB,CAAC;IAEJ,KAAK,CAAC,EAAc,EAAE,OAAgB;QACpC,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;QACxB,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,QAAQ,GAAG,IAAI,GAAG,CACtB,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAChF,CAAC;QACF,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC;YAAE,OAAO;QAChC,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC;YACtB,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC;YAC1C,GAAG,CAAC,EAAE,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;YACxD,GAAG,CAAC,EAAE,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC;SAC9D,CAAC,CAAC;QACH,MAAM,YAAY,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC/C,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,MAAM,IAAI,GAAG,CAAC,IAAY,EAAE,SAAkB,EAAW,EAAE;YACzD,IAAI,EAAE,KAAK,GAAG,IAAI;gBAAE,OAAO,KAAK,CAAC;YACjC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/C,MAAM,IAAI,GAAG,SAAS,IAAI,IAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,MAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC1D,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,KAAK,EAAE,EAAE;oBAClC,MAAM,OAAO,GAAG,MAAM,EAAE,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC;oBAC1C,IAAI,CAAC,IAAI,IAAI,CAAC,OAAO,IAAI,MAAO,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;wBAAE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACzE,CAAC,CAAC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBACpC,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,QAAQ,IAAI,EAAE;oBAAE,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YACxD,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAChC,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;gBAC9D,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,UAAU,EAAE,GAAG,CAAC,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;YACrF,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gBACrC,OAAO,CACL,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC;oBACvB,CAAC,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;oBAC9C,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC,CAC3C,CAAC;YACJ,CAAC;YACD,OAAO,cAAc,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QACnE,CAAC,CAAC;QACF,MAAM,IAAI,GAAG,CAAC,IAAY,EAAW,EAAE;YACrC,IAAI,EAAE,KAAK,GAAG,IAAI;gBAAE,OAAO,KAAK,CAAC;YACjC,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS;gBACzB,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;iBACvE,IAAI,IAAI,CAAC,IAAI,KAAK,OAAO;gBAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;iBACrD,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,YAAY,KAAK,IAAI;gBAC7D,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;iBAC5B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAChC,iEAAiE;gBACjE,qEAAqE;gBACrE,KAAK,MAAM,MAAM,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;oBAChC,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;wBAChC,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS;4BAAE,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBAC3D,CAAC;gBACH,CAAC;YACH,CAAC;iBAAM,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;gBAC/B,8DAA8D;gBAC9D,MAAM,YAAY,GAChB,IAAI,CAAC,IAAI,EAAE,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9E,KAAK,MAAM,IAAI,IAAI,YAAY;oBAAE,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS;wBAAE,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC1F,CAAC;YACD,OAAO,cAAc,CAAC,IAAI,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,IAAI,CAAC,CAAC;QACnE,CAAC,CAAC;QACF,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;YAAE,OAAO;QACjC,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC1B,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,YAAY,CAAC,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC;gBAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1E,CAAC;IACH,CAAC;IAED,IAAI,CACF,IAAwB,EACxB,EAAU,EACV,IAAY;QAEZ,wEAAwE;QACxE,kEAAkE;QAClE,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI,CAAC,SAAS,GAAG,EAAE;YAAE,OAAO,IAAI,CAAC;QAC7D,MAAM,GAAG,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;QAC/B,IACE,CAAC,GAAG;YACJ,IAAI,CAAC,IAAI,KAAK,KAAK;YACnB,IAAI,CAAC,IAAI,KAAK,MAAM;YACpB,IAAI,CAAC,IAAI,KAAK,YAAY;YAC1B,IAAI,CAAC,IAAI,KAAK,MAAM;YAEpB,OAAO,IAAI,CAAC;QACd,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;QACrB,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;QACjB,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACrB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,mBAAmB,CAAC,CAAC;QAC/C,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,sCAAsC,GAAG,EAAE,CAAC,CAAC;QACjE,IAAI,MAAM,GAAG,MAAM,EACjB,OAAO,GAAG,MAAM,CAAC;QACnB,IAAI,GAAG,EAAE,CAAC;YACR,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;YACjD,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC;YACzB,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC;QAC7B,CAAC;QACD,MAAM,IAAI,GAAG,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC,IAAI,CACJ,mBAAmB,IAAI,CAAC,QAAQ,YAAY,IAAI,SAAS,MAAM,SAAS,OAAO,4BAA4B,GAAG,EAAE,CACjH,CAAC;QACF,IAAI,CAAC,GAAG;YAAE,OAAO,EAAE,GAAG,EAAE,CAAC;QACzB,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;QACzB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,OAAO,0CAA0C,GAAG,gBAAgB,CAAC,CAAC;QAC7F,wEAAwE;QACxE,yEAAyE;QACzE,OAAO,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,CAAC;IACjD,CAAC;IAEO,cAAc,CAAC,EAAc;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QAC5C,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAC1B,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAU,CAAC;QACnC,oEAAoE;QACpE,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;YACrB,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI;YAChB,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;gBACb,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACxD,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,QAAQ,IAAI,EAAE;wBAAE,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACzD,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;SACF,CAAC,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACvC,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF"}

package/dist/backend/llvm/stack-unions.js

@@ -1,16 +1,21 @@
-import { isRefCounted, isUnitType, typeEquals } from "../../ir/ir.js";
+import { isRefCounted, isUnitType, typeEquals, } from "../../ir/ir.js";
 import { everyStmtList } from "../../ir/traverse.js";
 /** The ordinary tag and payload ABI can live on the stack when every
  * consumer is a projection. No RC or tracing entry point may receive this
  * box: its payload retains its ordinary independent owner instead. */
 export function canStackUnion(value, unions) {
-    if (value.kind !== "unionWrap" || value.type.kind !== "union" || value.type.unionId !== value.unionId)
+    if (value.kind !== "unionWrap" ||
+        value.type.kind !== "union" ||
+        value.type.unionId !== value.unionId)
         return false;
     const arm = unions.get(value.unionId)?.arms[value.tag];
     if (!arm || !typeEquals(arm, value.value.type))
         return false;
-    return isRefCounted(arm) || arm.kind === "f64" || arm.kind === "bool" || arm.kind === "procStream" ||
-        (isUnitType(arm) && value.value.kind === "unitLit");
+    return (isRefCounted(arm) ||
+        arm.kind === "f64" ||
+        arm.kind === "bool" ||
+        arm.kind === "procStream" ||
+        (isUnitType(arm) && value.value.kind === "unitLit"));
 }
 export function findLocalStackUnions(fn, lifetimes, unions) {
     const result = new Map();
@@ -20,7 +25,10 @@ export function findLocalStackUnions(fn, lifetimes, unions) {
     everyStmtList(fn.body, {
         expr: () => true,
         stmt: (stmt) => {
-            if (stmt.kind === "varDecl" && safe.has(stmt.localId) && stmt.init && canStackUnion(stmt.init, unions))
+            if (stmt.kind === "varDecl" &&
+                safe.has(stmt.localId) &&
+                stmt.init &&
+                canStackUnion(stmt.init, unions))
                 result.set(stmt.localId, stmt.init);
             return true;
         },
@@ -43,6 +51,7 @@ export function emitStackUnion(host, expr) {
         // the ordinary statement frame; a partial local never gains ownership.
         const value = host.emitExpr(expr.value);
         if (isRefCounted(type)) {
+            B.line(`store i64 0, ptr ${slot}`);
             B.line(`store ptr ${value.name}, ptr ${slot}`);
             payload = value;
         }

package/dist/backend/llvm/stack-unions.js.map

@@ -1 +1 @@
-{"version":3,"file":"stack-unions.js","sourceRoot":"","sources":["../../../src/backend/llvm/stack-unions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,UAAU,EAAiD,MAAM,gBAAgB,CAAC;AACrH,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAMrD;;sEAEsE;AACtE,MAAM,UAAU,aAAa,CAAC,KAAa,EAAE,MAAuC;IAClF,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO;QAAE,OAAO,KAAK,CAAC;IACpH,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvD,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7D,OAAO,YAAY,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY;QAChG,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,EAAc,EAAE,SAAwB,EAAE,MAAuC;IACpH,MAAM,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IAC5C,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,CAAC,IAAI,EAAE,IAAI;QAAE,OAAO,MAAM,CAAC;IAC/B,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;QACrB,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI;QAChB,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YACb,IAAI,IAAI,CAAC,IAAI,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,IAAI,IAAI,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC;gBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5I,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAUD,MAAM,UAAU,cAAc,CAAC,IAAwB,EAAE,IAAe;IACtE,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IACjB,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IACtD,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;IACjD,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,4CAA4C,GAAG,gBAAgB,CAAC,CAAC;IAC3F,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,IAAI,4CAA4C,GAAG,gBAAgB,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC7B,IAAI,OAAO,GAAmB,IAAI,CAAC;IACnC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,CAAC,CAAC,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;SAAM,CAAC;QACN,oEAAoE;QACpE,uEAAuE;QACvE,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;YAC/C,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,cAAc,KAAK,CAAC,IAAI,SAAS,CAAC,CAAC;YACjD,CAAC,CAAC,IAAI,CAAC,aAAa,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,CAAC,CAAC,IAAI,CAAC,gBAAgB,KAAK,CAAC,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IACD,CAAC,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,GAAG,SAAS,GAAG,EAAE,CAAC,CAAC;IAC5C,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;AAC/E,CAAC"}
+{"version":3,"file":"stack-unions.js","sourceRoot":"","sources":["../../../src/backend/llvm/stack-unions.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,YAAY,EACZ,UAAU,EACV,UAAU,GAIX,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,sBAAsB,CAAC;AAMrD;;sEAEsE;AACtE,MAAM,UAAU,aAAa,CAC3B,KAAa,EACb,MAAuC;IAEvC,IACE,KAAK,CAAC,IAAI,KAAK,WAAW;QAC1B,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO;QAC3B,KAAK,CAAC,IAAI,CAAC,OAAO,KAAK,KAAK,CAAC,OAAO;QAEpC,OAAO,KAAK,CAAC;IACf,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACvD,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC;QAAE,OAAO,KAAK,CAAC;IAC7D,OAAO,CACL,YAAY,CAAC,GAAG,CAAC;QACjB,GAAG,CAAC,IAAI,KAAK,KAAK;QAClB,GAAG,CAAC,IAAI,KAAK,MAAM;QACnB,GAAG,CAAC,IAAI,KAAK,YAAY;QACzB,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CACpD,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,EAAc,EACd,SAAwB,EACxB,MAAuC;IAEvC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAqB,CAAC;IAC5C,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;IAC3C,IAAI,CAAC,IAAI,EAAE,IAAI;QAAE,OAAO,MAAM,CAAC;IAC/B,aAAa,CAAC,EAAE,CAAC,IAAI,EAAE;QACrB,IAAI,EAAE,GAAG,EAAE,CAAC,IAAI;QAChB,IAAI,EAAE,CAAC,IAAI,EAAE,EAAE;YACb,IACE,IAAI,CAAC,IAAI,KAAK,SAAS;gBACvB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC;gBACtB,IAAI,CAAC,IAAI;gBACT,aAAa,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,CAAC;gBAEhC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC,CAAC;IACH,OAAO,MAAM,CAAC;AAChB,CAAC;AAUD,MAAM,UAAU,cAAc,CAAC,IAAwB,EAAE,IAAe;IACtE,MAAM,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;IACjB,MAAM,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,EAClB,GAAG,GAAG,CAAC,CAAC,IAAI,EAAE,EACd,IAAI,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC;IAClB,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,qBAAqB,CAAC,CAAC;IACjD,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,GAAG,4CAA4C,GAAG,gBAAgB,CAAC,CAAC;IAC3F,CAAC,CAAC,YAAY,CAAC,IAAI,CAAC,GAAG,IAAI,4CAA4C,GAAG,gBAAgB,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IAC7B,IAAI,OAAO,GAAmB,IAAI,CAAC;IACnC,IAAI,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC;QACrB,CAAC,CAAC,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;IACrC,CAAC;SAAM,CAAC;QACN,oEAAoE;QACpE,uEAAuE;QACvE,MAAM,KAAK,GAAG,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxC,IAAI,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;YACvB,CAAC,CAAC,IAAI,CAAC,oBAAoB,IAAI,EAAE,CAAC,CAAC;YACnC,CAAC,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;YAC/C,OAAO,GAAG,KAAK,CAAC;QAClB,CAAC;aAAM,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAChC,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,CAAC;YACrB,CAAC,CAAC,IAAI,CAAC,GAAG,IAAI,cAAc,KAAK,CAAC,IAAI,SAAS,CAAC,CAAC;YACjD,CAAC,CAAC,IAAI,CAAC,aAAa,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;QAC3C,CAAC;aAAM,CAAC;YACN,CAAC,CAAC,IAAI,CAAC,gBAAgB,KAAK,CAAC,IAAI,SAAS,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IACD,CAAC,CAAC,IAAI,CAAC,aAAa,IAAI,CAAC,GAAG,SAAS,GAAG,EAAE,CAAC,CAAC;IAC5C,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;AAC/E,CAAC"}

package/dist/backend/llvm/stack-unions.test.js

@@ -1,27 +1,62 @@
 import { expect, test } from "vitest";
-import { BOOL, F64, STRING, VOID, UNDEFINED_T, NULL_T, funcOf } from "../../ir/ir.js";
+import { BOOL, F64, STRING, VOID, UNDEFINED_T, NULL_T, funcOf, } from "../../ir/ir.js";
 import { validateModule } from "../../ir/validate.js";
 import { emitLlvmModule } from "./emitter.js";
 import { canStackUnion } from "./stack-unions.js";
 const loc = { file: "stack-unions.ts", start: 0, end: 0 };
 const record = { kind: "record", shapeId: "cell" };
 const optional = { kind: "union", unionId: "optional" };
-const ref = (localId, type = optional) => ({ kind: "varRef", localId, type, loc });
+const ref = (localId, type = optional) => ({
+    kind: "varRef",
+    localId,
+    type,
+    loc,
+});
 const num = (value = 1) => ({ kind: "numLit", value, type: F64, loc });
 const ret = (value) => ({ kind: "return", value, loc });
-const call = (callee, args, type = F64) => ({ kind: "call", callee, args, type, loc });
-const wrap = (value, tag = 0) => ({ kind: "unionWrap", value, tag, unionId: "optional", type: optional, loc });
-const narrow = (value, type) => ({ kind: "unionNarrow", value, unionId: "optional", tag: 0, type, loc });
+const call = (callee, args, type = F64) => ({
+    kind: "call",
+    callee,
+    args,
+    type,
+    loc,
+});
+const wrap = (value, tag = 0) => ({
+    kind: "unionWrap",
+    value,
+    tag,
+    unionId: "optional",
+    type: optional,
+    loc,
+});
+const narrow = (value, type) => ({
+    kind: "unionNarrow",
+    value,
+    unionId: "optional",
+    tag: 0,
+    type,
+    loc,
+});
 const effect = (expr) => ({ kind: "exprStmt", expr, loc });
-const fresh = () => ({ kind: "recordLit", fields: [{ name: "x", value: num(7) }], type: record, loc });
+const fresh = () => ({
+    kind: "recordLit",
+    fields: [{ name: "x", value: num(7) }],
+    type: record,
+    loc,
+});
 function fixture(arm = record) {
     const reader = {
-        name: "read", params: [{ localId: "value", name: "value", type: optional }], returnType: arm, loc,
+        name: "read",
+        params: [{ localId: "value", name: "value", type: optional }],
+        returnType: arm,
+        loc,
         locals: [{ id: "value", name: "value", type: optional, mutable: true }],
         body: [ret(narrow(ref("value"), arm))],
     };
     return {
-        irVersion: 13, sourceFile: loc.file, entry: "main",
+        irVersion: 14,
+        sourceFile: loc.file,
+        entry: "main",
         records: [{ id: "cell", fields: [{ name: "x", type: F64 }] }],
         unions: [{ id: "optional", arms: [arm, UNDEFINED_T, NULL_T] }],
         functions: [{ name: "main", params: [], returnType: VOID, locals: [], body: [], loc }, reader],
@@ -67,7 +102,13 @@ test("owned adapters release borrowed parameters while preserving returned paylo
 });
 test("plain function values retain the owned ABI", () => {
     const mod = fixture();
-    mod.functions[0].body.push(effect({ kind: "closure", fnName: "read", captures: [], type: funcOf([optional], record), loc }));
+    mod.functions[0].body.push(effect({
+        kind: "closure",
+        fnName: "read",
+        captures: [],
+        type: funcOf([optional], record),
+        loc,
+    }));
     const ir = emit(mod);
     const wrapper = body(ir, "sc_w_read");
     expect(wrapper).toContain("@sc_f_read(");
@@ -82,9 +123,16 @@ test("local construction releases its payload at lexical exit without freeing th
     expect(main).not.toMatch(/@scr_union_(?:new|retain|release)/);
     expect(main).toContain("@sc_rrelease_");
 });
-test.each(["return", "alias", "capture", "store", "mutable"])("keeps a %s consumer on the ordinary heap path", (reason) => {
+test.each(["return", "alias", "capture", "store"])("keeps a %s consumer on the ordinary heap path", (reason) => {
     const mod = fixture();
-    const work = { name: "work", params: [], locals: [], body: [], returnType: VOID, loc };
+    const work = {
+        name: "work",
+        params: [],
+        locals: [],
+        body: [],
+        returnType: VOID,
+        loc,
+    };
     mod.functions.unshift(work);
     declareLocal(mod, wrap(fresh()));
     if (reason === "return") {
@@ -97,24 +145,38 @@ test.each(["return", "alias", "capture", "store", "mutable"])("keeps a %s consum
     }
     if (reason === "capture") {
         work.locals[0].boxed = true;
-        mod.functions.push({ name: "lifted", captures: [{ localId: "item", name: "item", type: optional }], params: [],
-            locals: [{ id: "item", name: "item", type: optional, mutable: false, boxed: true }], body: [ret(ref("item"))], returnType: optional, loc });
-        work.body.push(effect({ kind: "closure", fnName: "lifted", captures: ["item"], type: funcOf([], optional), loc }));
+        mod.functions.push({
+            name: "lifted",
+            captures: [{ localId: "item", name: "item", type: optional }],
+            params: [],
+            locals: [{ id: "item", name: "item", type: optional, mutable: false, boxed: true }],
+            body: [ret(ref("item"))],
+            returnType: optional,
+            loc,
+        });
+        work.body.push(effect({
+            kind: "closure",
+            fnName: "lifted",
+            captures: ["item"],
+            type: funcOf([], optional),
+            loc,
+        }));
     }
     if (reason === "store") {
         mod.globals = [{ id: "%g.saved", name: "saved", type: optional, mutable: true }];
         work.body.push({ kind: "assign", localId: "%g.saved", value: ref("item"), loc });
     }
-    if (reason === "mutable")
-        work.locals[0].mutable = true;
     expect(body(emit(mod), "sc_f_work")).toContain("@scr_union_new_ref");
 });
 test("forwarded parameters never retain or release a borrowed stack box", () => {
     const mod = fixture();
     const outer = {
-        name: "outer", params: [{ localId: "value", name: "value", type: optional }],
+        name: "outer",
+        params: [{ localId: "value", name: "value", type: optional }],
         locals: [{ id: "value", name: "value", type: optional, mutable: true }],
-        body: [ret(call("read", [ref("value")], record))], returnType: record, loc,
+        body: [ret(call("read", [ref("value")], record))],
+        returnType: record,
+        loc,
     };
     mod.functions.push(outer);
     const ir = emit(mod), forwarded = body(ir, "sc_bf_outer");
@@ -128,8 +190,11 @@ test("mixed signatures transfer owned arguments and retain only borrowed snapsho
     fn.params.push({ localId: "saved", name: "saved", type: optional });
     fn.locals.push({ id: "saved", name: "saved", type: optional, mutable: true });
     fn.returnType = optional;
-    fn.body = [effect(narrow(ref("value"), record)),
-        { kind: "assign", localId: "saved", value: wrap(fresh()), loc }, ret(ref("saved"))];
+    fn.body = [
+        effect(narrow(ref("value"), record)),
+        { kind: "assign", localId: "saved", value: wrap(fresh()), loc },
+        ret(ref("saved")),
+    ];
     mod.functions[0].body.push(effect(call("read", [wrap(fresh()), wrap(fresh())], optional)));
     const ir = emit(mod), main = body(ir, "sc_f_main"), adapter = body(ir, "sc_f_read");
     expect(main.match(/call ptr @scr_union_new_ref/g)).toHaveLength(1);
@@ -160,11 +225,18 @@ test("mutable caller bindings take owned snapshots before later arguments", () =
     const fn = mod.functions[1];
     fn.params.push({ localId: "count", name: "count", type: F64 });
     fn.locals.push({ id: "count", name: "count", type: F64, mutable: true });
-    const second = { kind: "seqExpr", type: F64, loc,
-        stmts: [{ kind: "assign", localId: "item", value: wrap(fresh()), loc }], result: num() };
+    const second = {
+        kind: "seqExpr",
+        type: F64,
+        loc,
+        stmts: [{ kind: "assign", localId: "item", value: wrap(fresh()), loc }],
+        result: num(),
+    };
     mod.functions[0].body.push(effect(call("read", [ref("item"), second], record)));
     const main = body(emit(mod), "sc_f_main");
     expect(main).toContain("@scr_union_retain_v");
+    expect(main).toContain("@scr_union_new_ref");
+    expect(main).not.toContain("alloca %ScrUnion");
     expect(main.indexOf("@scr_union_retain_v")).toBeLessThan(main.indexOf("@sc_bf_read"));
     expect(main.lastIndexOf("@scr_union_release")).toBeGreaterThan(main.indexOf("@sc_bf_read"));
 });
@@ -192,10 +264,23 @@ test("scalar payloads initialize their complete ABI slot on both pointer widths"
     }
 });
 test("unit arms initialize tags and slots without acquiring payload ownership", () => {
-    for (const [unit, type, tag] of [["undefined", UNDEFINED_T, 1], ["null", NULL_T, 2]]) {
+    for (const [unit, type, tag] of [
+        ["undefined", UNDEFINED_T, 1],
+        ["null", NULL_T, 2],
+    ]) {
         const mod = fixture();
         mod.functions[1].returnType = BOOL;
-        mod.functions[1].body = [ret({ kind: "unionIsTag", value: ref("value"), unionId: "optional", tag, negated: false, type: BOOL, loc })];
+        mod.functions[1].body = [
+            ret({
+                kind: "unionIsTag",
+                value: ref("value"),
+                unionId: "optional",
+                tag,
+                negated: false,
+                type: BOOL,
+                loc,
+            }),
+        ];
         mod.functions[0].body.push(effect(call("read", [wrap({ kind: "unitLit", unit, type, loc }, tag)], BOOL)));
         const main = body(emit(mod), "sc_f_main");
         expect(main).toContain("store i64 0");
@@ -206,8 +291,17 @@ test("unit arms initialize tags and slots without acquiring payload ownership",
 test("checked payload errors retain caller cleanup and owned adapter cleanup", () => {
     const mod = fixture();
     const reader = mod.functions[1];
-    reader.body.unshift(effect({ kind: "libCall", fn: "error.nodeThrow", type: record, loc, args: [num(),
-            { kind: "strLit", value: "", type: STRING, loc }, { kind: "strLit", value: "failure", type: STRING, loc }] }));
+    reader.body.unshift(effect({
+        kind: "libCall",
+        fn: "error.nodeThrow",
+        type: record,
+        loc,
+        args: [
+            num(),
+            { kind: "strLit", value: "", type: STRING, loc },
+            { kind: "strLit", value: "failure", type: STRING, loc },
+        ],
+    }));
     mod.functions[0].body.push(effect(call("read", [wrap(fresh())], record)));
     const ir = emit(mod), main = body(ir, "sc_f_main");
     expect(main).toContain("@scr_exc_pending");
@@ -224,21 +318,46 @@ test("representation matching rejects absent, mismatched and effectful unit arms
     expect(canStackUnion(wrap(fresh(), 9), unions)).toBe(false);
     expect(canStackUnion(ref("value"), unions)).toBe(false);
     expect(canStackUnion(wrap(fresh()), new Map())).toBe(false);
-    const effectful = { kind: "seqExpr", stmts: [effect(num())],
-        result: { kind: "unitLit", unit: "undefined", type: UNDEFINED_T, loc }, type: UNDEFINED_T, loc };
+    const effectful = {
+        kind: "seqExpr",
+        stmts: [effect(num())],
+        result: { kind: "unitLit", unit: "undefined", type: UNDEFINED_T, loc },
+        type: UNDEFINED_T,
+        loc,
+    };
     expect(canStackUnion(wrap(effectful, 1), unions)).toBe(false);
 });
 test("adjacent throwing calls release argument snapshots before the next call", () => {
     const mod = fixture();
     const reader = mod.functions[1];
     reader.returnType = F64;
-    reader.body = [ret({ kind: "recordGet", obj: narrow(ref("value"), record), shapeId: "cell", field: "x", type: F64, loc })];
-    reader.body.unshift(effect({ kind: "libCall", fn: "error.nodeThrow", type: record, loc, args: [num(),
-            { kind: "strLit", value: "", type: STRING, loc }, { kind: "strLit", value: "failure", type: STRING, loc }] }));
+    reader.body = [
+        ret({
+            kind: "recordGet",
+            obj: narrow(ref("value"), record),
+            shapeId: "cell",
+            field: "x",
+            type: F64,
+            loc,
+        }),
+    ];
+    reader.body.unshift(effect({
+        kind: "libCall",
+        fn: "error.nodeThrow",
+        type: record,
+        loc,
+        args: [
+            num(),
+            { kind: "strLit", value: "", type: STRING, loc },
+            { kind: "strLit", value: "failure", type: STRING, loc },
+        ],
+    }));
     mod.globals = [{ id: "%g.item", name: "item", type: optional, mutable: true }];
     reader.body.unshift({ kind: "assign", localId: "%g.item", value: wrap(fresh()), loc });
     const calls = Array.from({ length: 80 }, () => call("read", [ref("%g.item")], F64));
-    mod.functions[0].body = [effect({ kind: "arrayLit", elems: calls, type: { kind: "array", elem: F64 }, loc })];
+    mod.functions[0].body = [
+        effect({ kind: "arrayLit", elems: calls, type: { kind: "array", elem: F64 }, loc }),
+    ];
     const main = body(emit(mod), "sc_f_main");
     expect(main.match(/call ptr @scr_union_retain_v/g)).toHaveLength(calls.length);
     // Each snapshot has one normal release and one exceptional release.