Modified the new type system to support generic differentiable types … (#2413)

* Modified the new type system to support generic differentiable types and added support for differentiating overloaded functions.

* Changed a few asserts to release asserts to avoid unreferenced variable errors

* Fixed a naming issue with TypeWitnessBreadcumb::Flavor::Decl

* Added logic to avoid tracking differentiable types if the module does not use auto-diff or define differentiable types.

* Moved the auto-diff passes to after the specialization step, added a more complex generics test

* Added a generics stress test and fixed AST-side logic. IR side needs some more work

* Added differential getter and setter logic, fixed multiple issues with DifferentiableTypeDictionary, added support for loops and conditions

* Changed differential getters to use pointer types, added getter type checking

* Fixed some bugs related to diff type registration and differential getters

* Removed some superfluous code

* Removed some more unused code.

* Fixed an issue with witness substitution

* Minor fix

Co-authored-by: Yong He <yonghe@outlook.com>

1 files changed, 17 insertions, 0 deletions

diff --git a/source/slang/slang-ast-builder.cpp b/source/slang/slang-ast-builder.cpp
index f8c208ac1..f6c550d69 100644
--- a/source/slang/slang-ast-builder.cpp
+++ b/source/slang/slang-ast-builder.cpp
@@ -179,6 +179,18 @@ Decl* SharedASTBuilder::findMagicDecl(const String& name)
     return m_magicDecls[name].GetValue();
 }
 
+Decl* SharedASTBuilder::tryFindMagicDecl(const String& name)
+{
+    if (m_magicDecls.ContainsKey(name))
+    {
+        return m_magicDecls[name].GetValue();
+    }
+    else
+    {
+        return nullptr;
+    }
+}
+
 // !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! ASTBuilder !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
 
 ASTBuilder::ASTBuilder(SharedASTBuilder* sharedASTBuilder, const String& name):
@@ -308,6 +320,11 @@ DeclRef<InterfaceDecl> ASTBuilder::getDifferentiableInterface()
     return declRef;
 }
 
+bool ASTBuilder::isDifferentiableInterfaceAvailable()
+{
+    return (m_sharedASTBuilder->tryFindMagicDecl("DifferentiableType") != nullptr);
+}
+
 DeclRef<Decl> ASTBuilder::getBuiltinDeclRef(const char* builtinMagicTypeName, Val* genericArg)
 {
     DeclRef<Decl> declRef;