diff options
Diffstat (limited to 'source/slang/slang-check-expr.cpp')
| -rw-r--r-- | source/slang/slang-check-expr.cpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/source/slang/slang-check-expr.cpp b/source/slang/slang-check-expr.cpp index 76af3694f..3c90c3ed8 100644 --- a/source/slang/slang-check-expr.cpp +++ b/source/slang/slang-check-expr.cpp @@ -2265,6 +2265,7 @@ namespace Slang { maybeRegisterDifferentiableType(m_astBuilder, arg->type.type); } + if (auto calleeExpr = as<DeclRefExpr>(checkedInvokeExpr->functionExpr)) { if (auto calleeDecl = as<FunctionDeclBase>(calleeExpr->declRef.getDecl())) @@ -2279,6 +2280,7 @@ namespace Slang newFuncExpr->type = checkedInvokeExpr->type; newFuncExpr->innerExpr = checkedInvokeExpr; newFuncExpr->loc = checkedInvokeExpr->loc; + newFuncExpr->flavor = TreatAsDifferentiableExpr::Flavor::Differentiable; checkedExpr = newFuncExpr; } else |