Update the type of a call inst during specialization. (#1569)

1 files changed, 6 insertions, 0 deletions

diff --git a/source/slang/slang-ir-specialize.cpp b/source/slang/slang-ir-specialize.cpp
index ceba4b03c..caaa700a0 100644
--- a/source/slang/slang-ir-specialize.cpp
+++ b/source/slang/slang-ir-specialize.cpp
@@ -901,6 +901,12 @@ struct SpecializationContext
         if(!calleeFunc)
             return;
 
+        // Update result type since the callee may have been changed.
+        if (inst->getDataType() != calleeFunc->getResultType())
+        {
+            inst->setFullType(calleeFunc->getResultType());
+        }
+
         // We can only specialize if we have access to a body for the callee.
         //
         if(!calleeFunc->isDefinition())