From 284cee1f246c072f190c87c8fb60c1d2181e458f Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 24 Apr 2023 19:44:23 -0700 Subject: Change AD checkpointing policy to recompute more. (#2836) Co-authored-by: Yong He --- source/slang/slang-ir-util.cpp | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'source/slang/slang-ir-util.cpp') diff --git a/source/slang/slang-ir-util.cpp b/source/slang/slang-ir-util.cpp index 03b74b36a..9348dfe8a 100644 --- a/source/slang/slang-ir-util.cpp +++ b/source/slang/slang-ir-util.cpp @@ -741,6 +741,24 @@ void moveParams(IRBlock* dest, IRBlock* src) } } +void removePhiArgs(IRInst* phiParam) +{ + auto block = cast(phiParam->getParent()); + UInt paramIndex = 0; + for (auto p = block->getFirstParam(); p; p = p->getNextParam()) + { + if (p == phiParam) + break; + paramIndex++; + } + for (auto predBlock : block->getPredecessors()) + { + auto termInst = as(predBlock->getTerminator()); + SLANG_ASSERT(paramIndex < termInst->getArgCount()); + termInst->removeArgument(paramIndex); + } +} + struct GenericChildrenMigrationContextImpl { IRCloneEnv cloneEnv; -- cgit v1.2.3