Dynamic dispatch bug fixes. (shader-slang#1541)

Co-authored-by: Yong He <yhe@nvidia.com>

Author: csyonghe

source/slang/slang-emit-cpp.cpp

@@ -1724,6 +1724,10 @@ void CPPSourceEmitter::emitRTTIObject(IRRTTIObject* rttiObject)
 void CPPSourceEmitter::_maybeEmitWitnessTableTypeDefinition(
     IRInterfaceType* interfaceType)
 {
+    if (m_interfaceTypesEmitted.Contains(interfaceType))
+        return;
+    m_interfaceTypesEmitted.Add(interfaceType);
+
     m_writer->emit("struct ");
     emitSimpleType(interfaceType);
     m_writer->emit("\n{\n");

source/slang/slang-emit-cpp.h

@@ -137,6 +137,8 @@ class CPPSourceEmitter: public CLikeSourceEmitter
 
     HashSet<const HLSLIntrinsic*> m_intrinsicEmitted;
 
+    HashSet<IRInterfaceType*> m_interfaceTypesEmitted;
+
     StringSlicePool m_slicePool;
 
     SemanticUsedFlags m_semanticUsedFlags;

source/slang/slang-ir-any-value-marshalling.cpp

@@ -126,13 +126,14 @@ namespace Slang
                 auto vectorType = static_cast<IRVectorType*>(dataType);
                 auto elementType = vectorType->getElementType();
                 auto elementCount = getIntVal(vectorType->getElementCount());
+                auto elementPtrType = builder->getPtrType(elementType);
                 for (IRIntegerValue i = 0; i < elementCount; i++)
                 {
-                    auto elementVal = builder->emitElementExtract(
-                        elementType,
+                    auto elementAddr = builder->emitElementAddress(
+                        elementPtrType,
                         concreteTypedVar,
                         builder->getIntValue(builder->getIntType(), i));
-                    emitMarshallingCode(builder, context, elementVal);
+                    emitMarshallingCode(builder, context, elementAddr);
                 }
                 break;
             }
@@ -469,6 +470,10 @@ namespace Slang
                 if (auto anyValueType = as<IRAnyValueType>(inst))
                     processAnyValueType(anyValueType);
             }
+            // Because we replaced all `AnyValueType` uses, some old type definitions (e.g. PtrType(AnyValueType))
+            // will become duplicates with new types we introduced (e.g. PtrType(AnyValueStruct)), and therefore
+            // invalidates our `globalValueNumberingMap` hash map. We need to rebuild it.
+            sharedContext->sharedBuilderStorage.deduplicateAndRebuildGlobalNumberingMap();
         }
     };
 

source/slang/slang-ir-deduplicate.cpp

@@ -17,6 +17,7 @@ namespace Slang
         IRInst* addConstantValue(IRConstant* value)
         {
             IRConstantKey key = { value };
+            value->setFullType((IRType*)addValue(value->getFullType()));
             if (auto newValue = builder->constantMap.TryGetValue(key))
                 return *newValue;
             builder->constantMap[key] = value;
@@ -34,15 +35,14 @@ namespace Slang
                 break;
             }
 
-            IRInstKey key = { value };
-            if (auto newValue = builder->globalValueNumberingMap.TryGetValue(key))
-                return *newValue;
-
             for (UInt i = 0; i < value->getOperandCount(); i++)
             {
                 value->setOperand(i, addValue(value->getOperand(i)));
             }
             value->setFullType((IRType*)addValue(value->getFullType()));
+            IRInstKey key = { value };
+            if (auto newValue = builder->globalValueNumberingMap.TryGetValue(key))
+                return *newValue;
             builder->globalValueNumberingMap[key] = value;
             return value;
         }
@@ -53,14 +53,14 @@ namespace Slang
         context.builder = this;
         bool changed = true;
         constantMap.Clear();
+        globalValueNumberingMap.Clear();
         for (auto inst : module->getGlobalInsts())
         {
             if (auto constVal = as<IRConstant>(inst))
             {
                 context.addConstantValue(constVal);
             }
         }
-        globalValueNumberingMap.Clear();
         List<IRInst*> instToRemove;
         for (auto inst : module->getGlobalInsts())
         {