call add_input_arguments on finish_ir

tgymnich · Sep 3, 2024 · c5f1475 · c5f1475
1 parent cbff0d4
commit c5f1475
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/metal.jl b/src/metal.jl
@@ -136,6 +136,9 @@ function finish_ir!(@nospecialize(job::CompilerJob{MetalCompilerTarget}), mod::L
 
     # add kernel metadata
     if job.config.kernel
+        add_input_arguments!(job, mod, entry)
+        entry = LLVM.functions(mod)[entry_fn]
+
         entry = add_address_spaces!(job, mod, entry)
 
         add_argument_metadata!(job, mod, entry)