2 files changed, 7 insertions, 0 deletions
diff --git a/source/compiler-core/slang-doc-extractor.cpp b/source/compiler-core/slang-doc-extractor.cpp
index ca947c94f..ffbfc9904 100644
--- a/source/compiler-core/slang-doc-extractor.cpp
+++ b/source/compiler-core/slang-doc-extractor.cpp
@@ -638,6 +638,12 @@ SlangResult DocMarkupExtractor::_findMarkup(const FindInfo& info, SearchStyle se
         {
             return _findMarkup(info, Location::Before, out);
         }
+        case SearchStyle::Attribute:
+        {
+            FindInfo newInfo = info;
+            newInfo.tokenIndex -= 2;
+            return _findMarkup(newInfo, Location::Before, out);
+        }
         case SearchStyle::Variable:
         {
             Location locs[] = { Location::Before, Location::AfterSemicolon };
diff --git a/source/compiler-core/slang-doc-extractor.h b/source/compiler-core/slang-doc-extractor.h
index fc486250e..30346ce6f 100644
--- a/source/compiler-core/slang-doc-extractor.h
+++ b/source/compiler-core/slang-doc-extractor.h
@@ -99,6 +99,7 @@ public:
         Before,             ///< Only allows before
         Function,           ///< Function/method
         GenericParam,       ///< Generic parameter
+        Attribute,          ///< Attribute definition
     };
 
         /// An input search item