Call llvm.reducer.register to register hyperobject struct members

OpenCilk · Dec 27, 2023 · a5c8b8e · a5c8b8e
1 parent aee817f
commit a5c8b8e
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/clang/lib/CodeGen/CGClass.cpp b/clang/lib/CodeGen/CGClass.cpp
@@ -715,6 +715,10 @@ void CodeGenFunction::EmitInitializerForField(FieldDecl *Field, LValue LHS,
   }
   }
 
+  ReducerCallbacks RCB = {0, 0};
+  if (getReducer(Field, RCB))
+    EmitReducerInit(Field, RCB, LHS.getPointer(*this));
+
   // Ensure that we destroy this object if an exception is thrown
   // later in the constructor.
   QualType::DestructionKind dtorKind = FieldType.isDestructedType();

diff --git a/clang/lib/CodeGen/CGDecl.cpp b/clang/lib/CodeGen/CGDecl.cpp
@@ -1816,7 +1816,8 @@ void CodeGenFunction::emitZeroOrPatternForAutoVarInit(QualType type,
   }
 }
 
-bool CodeGenFunction::getReducer(const VarDecl *D, ReducerCallbacks &CB) {
+bool CodeGenFunction::getReducer(const DeclaratorDecl *D,
+                                 ReducerCallbacks &CB) {
   if (const HyperobjectType *H = D->getType()->getAs<HyperobjectType>()) {
     if (H->hasCallbacks()) {
       CB.Identity = H->getIdentity();
@@ -1841,7 +1842,7 @@ void CodeGenFunction::destroyHyperobject(CodeGenFunction &CGF, Address Addr,
   }
 }
 
-void CodeGenFunction::EmitReducerInit(const VarDecl *D,
+void CodeGenFunction::EmitReducerInit(const DeclaratorDecl *D,
                                       const ReducerCallbacks &C,
                                       llvm::Value *Addr) {
   RValue Identity = EmitAnyExpr(C.Identity);

diff --git a/clang/lib/CodeGen/CodeGenFunction.h b/clang/lib/CodeGen/CodeGenFunction.h
@@ -3694,8 +3694,8 @@ class CodeGenFunction : public CodeGenTypeCache {
     Expr *Reduce;
   };
 
-  bool getReducer(const VarDecl *D, ReducerCallbacks &CB);
-  void EmitReducerInit(const VarDecl *D, const ReducerCallbacks &CB,
+  bool getReducer(const DeclaratorDecl *D, ReducerCallbacks &CB);
+  void EmitReducerInit(const DeclaratorDecl *D, const ReducerCallbacks &CB,
                        llvm::Value *Addr);
 
   /// Emits the alloca and debug information for the size expressions for each