From 0a81d11bc06e55089f7061225b9553329f697828 Mon Sep 17 00:00:00 2001 From: Yong He Date: Fri, 3 Sep 2021 01:57:31 -0700 Subject: Fix crash: dynamic dispatch of generic interface method. (#1929) * Fix crash: dynamic dispatch of generic interface method. * Fix memory error. Co-authored-by: Yong He --- source/slang/slang-lower-to-ir.cpp | 24 +++++++++++++++++++++--- 1 file changed, 21 insertions(+), 3 deletions(-) (limited to 'source/slang/slang-lower-to-ir.cpp') diff --git a/source/slang/slang-lower-to-ir.cpp b/source/slang/slang-lower-to-ir.cpp index 8a40dbde9..c1738887d 100644 --- a/source/slang/slang-lower-to-ir.cpp +++ b/source/slang/slang-lower-to-ir.cpp @@ -6702,9 +6702,18 @@ struct DeclLoweringVisitor : DeclVisitor { HashSet valuesToClone; markInstsToClone(valuesToClone, parentGeneric->getFirstBlock(), returnType); + // For Function Types, we always clone all generic parameters regardless of whether + // the generic parameter appears in the function signature or not. + if (returnType->getOp() == kIROp_FuncType) + { + for (auto genericParam : parentGeneric->getParams()) + { + markInstsToClone(valuesToClone, parentGeneric->getFirstBlock(), genericParam); + } + } if (valuesToClone.Count() == 0) { - // If returnType is independent of generic parameters, set + // If the new generic has no parameters, set // the generic inst's type to just `returnType`. parentGeneric->setFullType((IRType*)returnType); } @@ -6727,8 +6736,17 @@ struct DeclLoweringVisitor : DeclVisitor } IRInst* clonedReturnType = nullptr; cloneEnv.mapOldValToNew.TryGetValue(returnType, clonedReturnType); - SLANG_ASSERT(clonedReturnType); - typeBuilder.emitReturn(clonedReturnType); + if (clonedReturnType) + { + // If the type has explicit dependency on generic parameters, use + // the cloned type. + typeBuilder.emitReturn(clonedReturnType); + } + else + { + // Otherwise just use the original type value directly. + typeBuilder.emitReturn(returnType); + } parentGeneric->setFullType((IRType*)typeGeneric); returnType = typeGeneric; } -- cgit v1.2.3