diff options
| author | Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> | 2022-12-08 11:50:55 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-12-08 08:50:55 -0800 |
| commit | 468bb7ecf65c000c308adae511bf65a1ca4cc412 (patch) | |
| tree | 8042aaa77224d00f14a7267564ce7452ad6de67e /source/slang/slang-ir-autodiff-rev.cpp | |
| parent | 53e891eb28ceac5f956399c65f2ae27d37f3d724 (diff) | |
More type support for reverse-mode (#2551)
* Add vector arithmetic test. Make gradient accumulation work for any IRLoad
* Added support for general vector types, and split transposition into transpose & materialize to allow emitting the fully accumulated gradient for complex types.
* Several bug fixes + finished up support for vector & struct types + removed prop pass
* minor fixes (int/uint casts)
* Removed IRConstruct
* Added some type casts to prevent warnings
* minor fix for unused variable
Diffstat (limited to 'source/slang/slang-ir-autodiff-rev.cpp')
| -rw-r--r-- | source/slang/slang-ir-autodiff-rev.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/source/slang/slang-ir-autodiff-rev.cpp b/source/slang/slang-ir-autodiff-rev.cpp index 8ec8f581c..34a08ee93 100644 --- a/source/slang/slang-ir-autodiff-rev.cpp +++ b/source/slang/slang-ir-autodiff-rev.cpp @@ -511,7 +511,10 @@ struct BackwardDiffTranscriber this->makeParameterBlock(builder, as<IRFunc>(fwdDiffFunc)); // This steps adds a decoration to instructions that are computing the differential. - diffPropagationPass->propagateDiffInstDecoration(builder, fwdDiffFunc); + // TODO: This is disabled for now because fwd-mode already adds differential decorations + // wherever need. We need to run this pass only for user-writted forward derivativecode. + // + // diffPropagationPass->propagateDiffInstDecoration(builder, fwdDiffFunc); // Copy primal insts to the first block of the unzipped function, copy diff insts to the // second block of the unzipped function. |