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-ir-lower-generic-call.cpp | 5 ++--- source/slang/slang-lower-to-ir.cpp | 24 +++++++++++++++++++++--- 2 files changed, 23 insertions(+), 6 deletions(-) (limited to 'source') diff --git a/source/slang/slang-ir-lower-generic-call.cpp b/source/slang/slang-ir-lower-generic-call.cpp index 3571b7a27..7669593c6 100644 --- a/source/slang/slang-ir-lower-generic-call.cpp +++ b/source/slang/slang-ir-lower-generic-call.cpp @@ -218,9 +218,6 @@ namespace Slang auto newCall = builder->emitCallInst(calleeRetType, newCallee, args); auto callInstType = callInst->getDataType(); auto unpackInst = maybeUnpackValue(builder, callInstType, calleeRetType, newCall); - callInst->replaceUsesWith(unpackInst); - callInst->removeAndDeallocate(); - // Unpack other `out` arguments. for (auto& item : argsToUnpack) { @@ -229,6 +226,8 @@ namespace Slang auto unpackedVal = builder->emitUnpackAnyValue(originalValType, packedVal); builder->emitStore(item.dstArg, unpackedVal); } + callInst->replaceUsesWith(unpackInst); + callInst->removeAndDeallocate(); } IRInst* findInnerMostSpecializingBase(IRSpecialize* inst) 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