17e8765bdc
This PR fixes the compilation of structure projections with unboxed arguments marked `extern`, adding missing `dec` instructions. It led to leaking single allocations when such functions were used as closures or in the interpreter. This is the minimal working fix; `extern` should not replicate parts of the compilation pipeline, which will be possible via #10291.
38 lines
1 KiB
Text
38 lines
1 KiB
Text
/-
|
|
Copyright (c) 2025 Lean FRO LLC. All rights reserved.
|
|
Released under Apache 2.0 license as described in the file LICENSE.
|
|
Authors: Cameron Zwarich
|
|
-/
|
|
|
|
module
|
|
|
|
prelude
|
|
public import Lean.Compiler.IR.Boxing
|
|
import Lean.Compiler.IR.RC
|
|
|
|
public section
|
|
|
|
namespace Lean.IR
|
|
|
|
@[export lean_add_extern]
|
|
def addExtern (declName : Name) (externAttrData : ExternAttrData) : CoreM Unit := do
|
|
let mut type ← Compiler.LCNF.getOtherDeclMonoType declName
|
|
let mut params := #[]
|
|
let mut nextVarIndex := 0
|
|
repeat
|
|
let .forallE _ d b _ := type | break
|
|
let borrow := isMarkedBorrowed d
|
|
let ty ← toIRType d
|
|
params := params.push { x := ⟨nextVarIndex⟩, borrow, ty }
|
|
type := b
|
|
nextVarIndex := nextVarIndex + 1
|
|
let irType ← toIRType type
|
|
let decls := #[.extern declName params irType externAttrData]
|
|
if !isPrivateName declName then
|
|
modifyEnv (Compiler.LCNF.setDeclPublic · declName)
|
|
let decls := ExplicitBoxing.addBoxedVersions (← Lean.getEnv) decls
|
|
let decls ← explicitRC decls
|
|
logDecls `result decls
|
|
addDecls decls
|
|
|
|
end Lean.IR
|