From 368ec3116ea0f10f44acbf76b5dc9e34d6ff3d32 Mon Sep 17 00:00:00 2001 From: Edward Liu Date: Mon, 14 Nov 2022 12:08:01 -0800 Subject: Minimum binary arithmetic reverse autodiff working. (#2514) * Initial plumbing of backward autodiff in the frontend. * More plumbing. * Initial reverse autodiff working. * Bug fixes. * Misc. * Remove redundant code. * More clean up. * Misc. * Rebase and add backward diff test. * Disable test. * Clean up. * Minor fix. Co-authored-by: Yong He --- source/slang/slang-emit.cpp | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) (limited to 'source/slang/slang-emit.cpp') diff --git a/source/slang/slang-emit.cpp b/source/slang/slang-emit.cpp index de9d23e97..2478eccc2 100644 --- a/source/slang/slang-emit.cpp +++ b/source/slang/slang-emit.cpp @@ -376,11 +376,7 @@ Result linkAndOptimizeIR( dumpIRIfEnabled(codeGenContext, irModule, "BEFORE-AUTODIFF"); // Process higher-order calles to auto-diff passes. - // 1. Generate JVP code wherever necessary. (Linearization or "forward-mode" pass) - processForwardDifferentiableFuncs(irModule, sink); - - // 2. Transpose JVP to VJP code wherever needed. (Transposition or "reverse-mode" pass) - // processVJPDerivativeMarkers(module); // Disabled currently. No impl yet. + processDifferentiableFuncs(irModule, sink); stripAutoDiffDecorations(irModule); -- cgit v1.2.3