From 7f64b2a9e3eb7aea13de550bd24c1aea7787c94b Mon Sep 17 00:00:00 2001 From: Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> Date: Wed, 4 Jan 2023 23:40:13 +0530 Subject: Multi-block reverse-mode autodiff (#2576) * Initial multi-block implementation * Implemented multi-block reverse-mode (without loops) * Added logic to remove block-level decorations to avoid confusing IR simplification passes * Fixed issues with block-level decorations during IR simplification by removing them prior to simplification. Co-authored-by: Yong He --- source/slang/slang-ir.cpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'source/slang/slang-ir.cpp') diff --git a/source/slang/slang-ir.cpp b/source/slang/slang-ir.cpp index 33130cfb3..d8a8fb7c4 100644 --- a/source/slang/slang-ir.cpp +++ b/source/slang/slang-ir.cpp @@ -6623,6 +6623,10 @@ namespace Slang case kIROp_Reinterpret: case kIROp_GetNativePtr: return false; + + case kIROp_ForwardDifferentiate: + case kIROp_BackwardDifferentiate: + return false; } } -- cgit v1.2.3