summaryrefslogtreecommitdiff
path: root/source/slang/slang-check-decl.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'source/slang/slang-check-decl.cpp')
-rw-r--r--source/slang/slang-check-decl.cpp22
1 files changed, 20 insertions, 2 deletions
diff --git a/source/slang/slang-check-decl.cpp b/source/slang/slang-check-decl.cpp
index cb3db9e39..f25821dac 100644
--- a/source/slang/slang-check-decl.cpp
+++ b/source/slang/slang-check-decl.cpp
@@ -1656,8 +1656,7 @@ namespace Slang
RequirementWitness witnessValue;
auto requirementDecl = m_astBuilder->getSharedASTBuilder()->findBuiltinRequirementDecl(BuiltinRequirementKind::DifferentialType);
if (!inheritanceDecl->witnessTable->getRequirementDictionary().tryGetValue(requirementDecl, witnessValue))
- return;
-
+ return;
// A type used as differential type must have itself as its own differential type.
if (witnessValue.getFlavor() != RequirementWitness::Flavor::val)
return;
@@ -5781,6 +5780,16 @@ namespace Slang
interfaceDecl->members.add(reqDecl);
reqDecl->parentDecl = interfaceDecl;
+ if (!decl->hasModifier<NoDiffThisAttribute>())
+ {
+ // Build decl-ref-type from interface.
+ auto interfaceType = DeclRefType::create(getASTBuilder(), makeDeclRef(interfaceDecl));
+
+ // If the interface is differentiable, make the this type a pair.
+ if (tryGetDifferentialType(getASTBuilder(), interfaceType))
+ reqDecl->diffThisType = getDifferentialPairType(interfaceType);
+ }
+
auto reqRef = m_astBuilder->create<DerivativeRequirementReferenceDecl>();
reqRef->referencedDecl = reqDecl;
reqRef->parentDecl = decl;
@@ -5800,6 +5809,15 @@ namespace Slang
setFuncTypeIntoRequirementDecl(reqDecl, diffFuncType);
interfaceDecl->members.add(reqDecl);
reqDecl->parentDecl = interfaceDecl;
+ if (!decl->hasModifier<NoDiffThisAttribute>())
+ {
+ // Build decl-ref-type from interface.
+ auto interfaceType = DeclRefType::create(getASTBuilder(), makeDeclRef(interfaceDecl));
+
+ // If the interface is differentiable, make the this type a pair.
+ if (tryGetDifferentialType(getASTBuilder(), interfaceType))
+ reqDecl->diffThisType = getDifferentialPairType(interfaceType);
+ }
auto reqRef = m_astBuilder->create<DerivativeRequirementReferenceDecl>();
reqRef->referencedDecl = reqDecl;