1 files changed, 4 insertions, 6 deletions
diff --git a/source/slang/slang-ir.cpp b/source/slang/slang-ir.cpp
index fd5ea0fc7..696e862d6 100644
--- a/source/slang/slang-ir.cpp
+++ b/source/slang/slang-ir.cpp
@@ -5075,14 +5075,12 @@ namespace Slang
             getIntValue(getIntType(), (IRIntegerValue)mode));
     }
 
-    IRInst* IRBuilder::addNumThreadsDecoration(IRInst* inst, Int x, Int y, Int z)
+    IRInst* IRBuilder::addNumThreadsDecoration(IRInst* inst, IRInst* x, IRInst* y, IRInst* z)
     {
-        IRType* intType = getIntType();
-
         IRInst* operands[3] = {
-            getIntValue(intType, x),
-            getIntValue(intType, y),
-            getIntValue(intType, z)
+            x,
+            y,
+            z
         };
 
         return addDecoration(inst, kIROp_NumThreadsDecoration, operands, 3);