diff options
| author | Yong He <yonghe@outlook.com> | 2023-02-01 14:18:57 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-02-01 14:18:57 -0800 |
| commit | bbd1e1786401bb88c34802b987d4da72e2364503 (patch) | |
| tree | 99a4be95ae517fd710fc032a1debdac917dd3ac2 /source/slang/slang-ir-autodiff.cpp | |
| parent | c5895fb0b82fd14fbe45b58d5fc7f75d67625d15 (diff) | |
Support `out` parameters in backward differentiation. (#2619)
* Support `out` parameters in backward differentiation.
* Fixes.
* Fix cleanup.
---------
Co-authored-by: Yong He <yhe@nvidia.com>
Diffstat (limited to 'source/slang/slang-ir-autodiff.cpp')
| -rw-r--r-- | source/slang/slang-ir-autodiff.cpp | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/source/slang/slang-ir-autodiff.cpp b/source/slang/slang-ir-autodiff.cpp index 8952f9756..7a2e8c75e 100644 --- a/source/slang/slang-ir-autodiff.cpp +++ b/source/slang/slang-ir-autodiff.cpp @@ -24,7 +24,7 @@ bool isBackwardDifferentiableFunc(IRInst* func) return false; } -static IRInst* _lookupWitness(IRBuilder* builder, IRInst* witness, IRInst* requirementKey) +IRInst* _lookupWitness(IRBuilder* builder, IRInst* witness, IRInst* requirementKey) { if (auto witnessTable = as<IRWitnessTable>(witness)) { @@ -400,6 +400,14 @@ IRInst* DifferentiableTypeConformanceContext::lookUpInterfaceMethod(IRBuilder* b return nullptr; } +IRInst* DifferentiableTypeConformanceContext::getDifferentialTypeFromDiffPairType( + IRBuilder* builder, IRDifferentialPairType* diffPairType) +{ + auto witness = diffPairType->getWitness(); + SLANG_RELEASE_ASSERT(witness); + return _lookupWitness(builder, witness, sharedContext->differentialAssocTypeStructKey); +} + void DifferentiableTypeConformanceContext::buildGlobalWitnessDictionary() { for (auto globalInst : sharedContext->moduleInst->getChildren()) |