From a911ca6e06ce41e403b80fe6054162393491c8ac Mon Sep 17 00:00:00 2001 From: Yong He Date: Mon, 13 Mar 2023 10:57:28 -0700 Subject: Support high order diff pattern: `bwd_diff(fwd_diff(f))`. (#2695) * Support high order diff pattern: `bwd_diff(fwd_diff(f))`. * Fix. --------- Co-authored-by: Yong He --- source/slang/slang-ir-check-differentiability.cpp | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) (limited to 'source/slang/slang-ir-check-differentiability.cpp') diff --git a/source/slang/slang-ir-check-differentiability.cpp b/source/slang/slang-ir-check-differentiability.cpp index 14f6394e2..6f97ce076 100644 --- a/source/slang/slang-ir-check-differentiability.cpp +++ b/source/slang/slang-ir-check-differentiability.cpp @@ -73,16 +73,13 @@ public: bool isDifferentiableFunc(IRInst* func, DifferentiableLevel level) { - if (level == DifferentiableLevel::Forward) + switch (func->getOp()) { - switch (func->getOp()) - { - case kIROp_ForwardDifferentiate: - case kIROp_BackwardDifferentiate: - return true; - default: - break; - } + case kIROp_ForwardDifferentiate: + case kIROp_BackwardDifferentiate: + return isDifferentiableFunc(func->getOperand(0), level); + default: + break; } func = getResolvedInstForDecorations(func); -- cgit v1.2.3