call add_input_arguments on finish_ir

tgymnich · tgymnich · commit c5f14753597e · 2024-09-03T13:31:23.000+02:00
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)
