From 134dd7eb26fc7988ae13559d276cbf337b4b9d27 Mon Sep 17 00:00:00 2001 From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> Date: Mon, 30 Jan 2023 11:46:36 -0500 Subject: Overhauled reverse-mode control flow handling (#2608) * Added switch-case support; fixed non-diff parameter transposition * Made region propagation much more robust. Partial loop unzip implementation * WIP: Added most loop handling code, and a test. Still untested * Added CFG Normalization pass + CFG Reversal Pass + Loop Unzipping + most loop transcription * Add single-iter-loop test. * proj files * removed comments * Update reverse-loop.slang * Removed out-of-date code * Disabled IR validation during constructSSA phase of normalizeCFG. constructSSA now reuses sharedBuilder * Moved normalizeCFG() call to prepareFuncForBackwardDiff() --- source/slang/slang-ir.cpp | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) (limited to 'source/slang/slang-ir.cpp') diff --git a/source/slang/slang-ir.cpp b/source/slang/slang-ir.cpp index 845232ae6..e72ba8c9f 100644 --- a/source/slang/slang-ir.cpp +++ b/source/slang/slang-ir.cpp @@ -4764,6 +4764,32 @@ namespace Slang return inst; } + IRInst* IRBuilder::emitLoop( + IRBlock* target, + IRBlock* breakBlock, + IRBlock* continueBlock, + Int argCount, + IRInst*const* args) + { + List argList; + + argList.add(target); + argList.add(breakBlock); + argList.add(continueBlock); + + for (Count ii = 0; ii < argCount; ii++) + argList.add(args[ii]); + + auto inst = createInst( + this, + kIROp_loop, + nullptr, + argList.getCount(), + argList.getBuffer()); + addInst(inst); + return inst; + } + IRInst* IRBuilder::emitBranch( IRInst* val, IRBlock* trueBlock, -- cgit v1.2.3