From 1c9b33157322751c456bf7abbd386edccf4413c3 Mon Sep 17 00:00:00 2001 From: Yong He Date: Sat, 14 Jan 2023 22:50:57 -0800 Subject: Support custom backward derivative attribute. (#2594) --- source/slang/slang-ir-check-differentiability.cpp | 15 +-------------- 1 file changed, 1 insertion(+), 14 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 83351d07b..8413e7e79 100644 --- a/source/slang/slang-ir-check-differentiability.cpp +++ b/source/slang/slang-ir-check-differentiability.cpp @@ -71,6 +71,7 @@ public: { case kIROp_ForwardDerivativeDecoration: case kIROp_ForwardDifferentiableDecoration: + case kIROp_UserDefinedBackwardDerivativeDecoration: case kIROp_BackwardDerivativeDecoration: case kIROp_BackwardDifferentiableDecoration: return true; @@ -140,20 +141,6 @@ public: return false; } - bool isBackwardDifferentiableFunc(IRInst* func) - { - for (auto decorations : func->getDecorations()) - { - switch (decorations->getOp()) - { - case kIROp_BackwardDerivativeDecoration: - case kIROp_BackwardDifferentiableDecoration: - return true; - } - } - return false; - } - bool isDifferentiableType(DifferentiableTypeConformanceContext& context, IRInst* typeInst) { HashSet processedSet; -- cgit v1.2.3