Visit statment children of HyperobjectType

OpenCilk · Jul 13, 2022 · 3cbba61 · 3cbba61
1 parent 6f78d2a
commit 3cbba61
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/clang/include/clang/AST/ASTNodeTraverser.h b/clang/include/clang/AST/ASTNodeTraverser.h
@@ -340,6 +340,10 @@ class ASTNodeTraverser
 
   void VisitHyperobjectType(const HyperobjectType *T) {
     Visit(T->getElementType());
+    if (T->getIdentity())
+      Visit(T->getIdentity());
+    if (T->getReduce())
+      Visit(T->getReduce());
   }
   void VisitComplexType(const ComplexType *T) { Visit(T->getElementType()); }
   void VisitLocInfoType(const LocInfoType *T) {

diff --git a/clang/include/clang/AST/RecursiveASTVisitor.h b/clang/include/clang/AST/RecursiveASTVisitor.h
@@ -895,6 +895,10 @@ DEF_TRAVERSE_TYPE(ComplexType, { TRY_TO(TraverseType(T->getElementType())); })
 
 DEF_TRAVERSE_TYPE(HyperobjectType, {
     TRY_TO(TraverseType(T->getElementType()));
+    if (Stmt *I = T->getIdentity())
+      TRY_TO(TraverseStmt(I));
+    if (Stmt *R = T->getReduce())
+      TRY_TO(TraverseStmt(R));
   })
 
 DEF_TRAVERSE_TYPE(PointerType, { TRY_TO(TraverseType(T->getPointeeType())); })
@@ -1138,7 +1142,12 @@ DEF_TRAVERSE_TYPELOC(ComplexType, {
 })
 
 DEF_TRAVERSE_TYPELOC(HyperobjectType, {
-  TRY_TO(TraverseType(TL.getTypePtr()->getElementType()));
+  const HyperobjectType *H = TL.getTypePtr();
+  TRY_TO(TraverseType(H->getElementType()));
+  if (Stmt *I = H->getIdentity())
+    TRY_TO(TraverseStmt(I));
+  if (Stmt *R = H->getReduce())
+    TRY_TO(TraverseStmt(R));
 })
 
 DEF_TRAVERSE_TYPELOC(PointerType,