Add support for `[PrimalSubstitute]` and `[PrimalSubstituteOf]`. (#2691)

* Add support for `[PrimalSubstitute]` and `[PrimalSubstituteOf]`.

* Fix

* Fix.

* Cleanup.

---------

Co-authored-by: Yong He <yhe@nvidia.com>

1 files changed, 4 insertions, 1 deletions

diff --git a/source/slang/slang-check-impl.h b/source/slang/slang-check-impl.h
index fc1b622cc..3d40b10e9 100644
--- a/source/slang/slang-check-impl.h
+++ b/source/slang/slang-check-impl.h
@@ -287,10 +287,11 @@ namespace Slang
 
         void registerAssociatedDecl(Decl* original, DeclAssociationKind assoc, Decl* declaration);
 
-        List<DeclAssociation> const& getAssociatedDeclsForDecl(Decl* decl);
+        List<RefPtr<DeclAssociation>> const& getAssociatedDeclsForDecl(Decl* decl);
 
         bool isDifferentiableFunc(FunctionDeclBase* func);
         bool isBackwardDifferentiableFunc(FunctionDeclBase* func);
+        FunctionDifferentiableLevel _getFuncDifferentiableLevelImpl(FunctionDeclBase* func, int recurseLimit);
         FunctionDifferentiableLevel getFuncDifferentiableLevel(FunctionDeclBase* func);
         
     private:
@@ -1951,6 +1952,8 @@ namespace Slang
 
         Expr* visitForwardDifferentiateExpr(ForwardDifferentiateExpr* expr);
         Expr* visitBackwardDifferentiateExpr(BackwardDifferentiateExpr* expr);
+        Expr* visitPrimalSubstituteExpr(PrimalSubstituteExpr* expr);
+
         Expr* visitTreatAsDifferentiableExpr(TreatAsDifferentiableExpr* expr);
 
         Expr* visitGetArrayLengthExpr(GetArrayLengthExpr* expr);