Update ModuleCodegen.cs

Draco-lang · Dec 16, 2023 · 213a498 · 213a498
1 parent 3bb41d0
commit 213a498
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/Draco.Compiler/Internal/OptimizingIr/ModuleCodegen.cs b/src/Draco.Compiler/Internal/OptimizingIr/ModuleCodegen.cs
@@ -79,13 +79,13 @@ public override void VisitGlobal(GlobalSymbol globalSymbol)
 
     public override void VisitFunction(FunctionSymbol functionSymbol)
     {
-        if (functionSymbol is not SourceFunctionSymbol sourceFunction) return;
+        if (functionSymbol.Body is null) return;
 
         // Add procedure
         var procedure = this.module.DefineProcedure(functionSymbol);
 
         // Create the body
-        var body = this.RewriteBody(sourceFunction.Body);
+        var body = this.RewriteBody(functionSymbol.Body);
         // Yank out potential local functions and closures
         var (bodyWithoutLocalFunctions, localFunctions) = ClosureRewriter.Rewrite(body);
         // Compile it