From 86fc50c5092fbccf6072dcf7bbdfafb8915f02c8 Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Wed, 8 Mar 2023 21:52:34 -0800
Subject: Add support for `[PrimalSubstitute]` and `[PrimalSubstituteOf]`.
 (#2691)

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

* Fix

* Fix.

* Cleanup.

---------

Co-authored-by: Yong He <yhe@nvidia.com>
---
 source/slang/slang-check-overload.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'source/slang/slang-check-overload.cpp')

diff --git a/source/slang/slang-check-overload.cpp b/source/slang/slang-check-overload.cpp
index 91af731ad..f786089f8 100644
--- a/source/slang/slang-check-overload.cpp
+++ b/source/slang/slang-check-overload.cpp
@@ -1549,7 +1549,7 @@ namespace Slang
                 // Base is a normal or fully specialized generic function.
                 OverloadCandidate candidate;
                 candidate.flavor = OverloadCandidate::Flavor::Expr;
-                if (auto diffExpr = as<DifferentiateExpr>(expr))
+                if (auto diffExpr = as<HigherOrderInvokeExpr>(expr))
                 {
                     candidate.funcType = as<FuncType>(diffExpr->type.type);
                 }
-- 
cgit v1.2.3