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-check-expr.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-check-expr.cpp')
| -rw-r--r-- | source/slang/slang-check-expr.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/source/slang/slang-check-expr.cpp b/source/slang/slang-check-expr.cpp index 336682bf4..1ffc45fbd 100644 --- a/source/slang/slang-check-expr.cpp +++ b/source/slang/slang-check-expr.cpp @@ -2136,9 +2136,9 @@ namespace Slang type->paramTypes.add(derivType); } } - + // Last parameter is the initial derivative of the original return type - type->paramTypes.add(originalType->resultType); + type->paramTypes.add(getDifferentialType(m_astBuilder, originalType->resultType, SourceLoc())); return type; } |