From 24df5515d6c2f8537683d0e48d27a161c394e7cd Mon Sep 17 00:00:00 2001
From: Yong He <yonghe@outlook.com>
Date: Fri, 30 Aug 2024 16:32:34 -0700
Subject: Support extension on generic type. (#4968)

---
 source/slang/slang-parser.cpp | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

(limited to 'source/slang/slang-parser.cpp')

diff --git a/source/slang/slang-parser.cpp b/source/slang/slang-parser.cpp
index 04ada006c..121c68be7 100644
--- a/source/slang/slang-parser.cpp
+++ b/source/slang/slang-parser.cpp
@@ -3335,12 +3335,15 @@ namespace Slang
 
     static NodeBase* parseExtensionDecl(Parser* parser, void* /*userData*/)
     {
-        ExtensionDecl* decl = parser->astBuilder->create<ExtensionDecl>();
-        parser->FillPosition(decl);
-        decl->targetType = parser->ParseTypeExp();
-        parseOptionalInheritanceClause(parser, decl);
-        parseDeclBody(parser, decl);
-        return decl;
+        return parseOptGenericDecl(parser, [&](GenericDecl*)
+            {
+                ExtensionDecl* decl = parser->astBuilder->create<ExtensionDecl>();
+                parser->FillPosition(decl);
+                decl->targetType = parser->ParseTypeExp();
+                parseOptionalInheritanceClause(parser, decl);
+                parseDeclBody(parser, decl);
+                return decl;
+            });
     }
 
 
-- 
cgit v1.2.3