diff options
| author | venkataram-nv <vedavamadath@nvidia.com> | 2024-09-18 20:42:07 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-18 20:42:07 -0700 |
| commit | b808aa4df50d46eaa569561f7e464c55c1c2d72a (patch) | |
| tree | 5483a3f9e73a401ff82d66fd1ac3729a9a84a97c /source/slang/slang-ir-autodiff-transcriber-base.cpp | |
| parent | 3240799c00488858afc7eeac9d1dc479609a1040 (diff) | |
Report AD checkpoint contexts (#5058)
* Transferring source locations when creating phi instructions
* Tracking for simple variables
* Deriving source locations for loop counters
* Printing checkpoint structure breakdown
* More readable output format
* Special behavior for loop counters
* Writing report to file
* Add slangc option to enable checkpoint reports
* Display types of checkpointed fields
* Message in case there are no checkpointing contexts
* Catch source locations for function calls
* Source cleanup
* Fix compilation warnings
* Remove stray dump()
* Provide the report through diagnostic notes
* Add missing path for sourceLoc during unzip pass
* Add tests for reporting intermediates
* Include more transfer cases for source locations
* Fix ordering in address elimination
* Fill in more holes with source location transfer
* Remove debugging line
* Reverting changes to diagnostic sink
* Simplify address elimination using source location RAII contexts
* Eliminating manual source loc transfers in forward transcription
* Fix local var adaptation to use RAII location setter
* Simplify primal hoisting logic for source location transfer
* Simplify unzipping with RAII location scopes
* Simplify transpose logic
* Cleaning up for rev.cpp
* Reverting spacing changes
* Fix mistake with source loc RAII instantiation
* Fix formatting issues
Diffstat (limited to 'source/slang/slang-ir-autodiff-transcriber-base.cpp')
| -rw-r--r-- | source/slang/slang-ir-autodiff-transcriber-base.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/source/slang/slang-ir-autodiff-transcriber-base.cpp b/source/slang/slang-ir-autodiff-transcriber-base.cpp index da69ed8ae..1fa76c730 100644 --- a/source/slang/slang-ir-autodiff-transcriber-base.cpp +++ b/source/slang/slang-ir-autodiff-transcriber-base.cpp @@ -1033,8 +1033,9 @@ InstPair AutoDiffTranscriberBase::transcribeInst(IRBuilder* builder, IRInst* ori if (as<IRModuleInst>(origInst->getParent()) && !as<IRType>(origInst)) return InstPair(origInst, nullptr); - auto result = transcribeInstImpl(builder, origInst); + IRBuilderSourceLocRAII sourceLocationScope(builder, origInst->sourceLoc); + auto result = transcribeInstImpl(builder, origInst); if (result.primal == nullptr && result.differential == nullptr) { if (auto origType = as<IRType>(origInst)) |