From c5f14753597e05526375cd58069b45a7715a2ca9 Mon Sep 17 00:00:00 2001
From: Tim Gymnich <tgymnich@icloud.com>
Date: Tue, 3 Sep 2024 13:31:23 +0200
Subject: [PATCH] call add_input_arguments on finish_ir

---
 src/metal.jl | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/metal.jl b/src/metal.jl
index 87d2106e..f2867b67 100644
--- 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)