Capability type checking. (#3530)

* Capability type checking.

* Fix.

---------

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

1 files changed, 1 insertions, 0 deletions

diff --git a/source/slang/slang-parser.cpp b/source/slang/slang-parser.cpp
index c5007569e..3940e59cf 100644
--- a/source/slang/slang-parser.cpp
+++ b/source/slang/slang-parser.cpp
@@ -4900,6 +4900,7 @@ namespace Slang
                         parser->sink->diagnose(caseName.loc, Diagnostics::unknownTargetName, caseName.getContent());
                     }
                     targetCase->capability = int32_t(cap);
+                    targetCase->capabilityToken = caseName;
                     targetCase->loc = caseName.loc;
                     targetCase->body = bodyStmt;
                     stmt->targetCases.add(targetCase);