Support extension on generic type. (#4968)

1 files changed, 9 insertions, 6 deletions

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;
+            });
     }