Lines Matching refs:ReductionInfos

2372     ArrayRef<ReductionInfo> ReductionInfos, Value *SrcBase, Value *DestBase,  in emitReductionListCopy()  argument
2380 for (auto En : enumerate(ReductionInfos)) { in emitReductionListCopy()
2484 const LocationDescription &Loc, ArrayRef<ReductionInfo> ReductionInfos, in emitInterWarpCopyFunction() argument
2560 for (auto En : enumerate(ReductionInfos)) { in emitInterWarpCopyFunction()
2617 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitInterWarpCopyFunction()
2713 ArrayRef<ReductionInfo> ReductionInfos, Function *ReduceFn, in emitShuffleAndReduceFunction() argument
2756 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitShuffleAndReduceFunction()
2794 AllocaIP, CopyAction::RemoteLaneToThread, RedListArrayTy, ReductionInfos, in emitShuffleAndReduceFunction()
2864 ReductionInfos, RemoteListAddrCast, ReduceList); in emitShuffleAndReduceFunction()
2878 ArrayRef<ReductionInfo> ReductionInfos, Type *ReductionsBufferTy, in emitListToGlobalCopyFunction() argument
2930 for (auto En : enumerate(ReductionInfos)) { in emitListToGlobalCopyFunction()
2933 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitListToGlobalCopyFunction()
2988 ArrayRef<ReductionInfo> ReductionInfos, Function *ReduceFn, in emitListToGlobalReduceFunction() argument
3021 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitListToGlobalReduceFunction()
3048 for (auto En : enumerate(ReductionInfos)) { in emitListToGlobalReduceFunction()
3071 ArrayRef<ReductionInfo> ReductionInfos, Type *ReductionsBufferTy, in emitGlobalToListCopyFunction() argument
3121 for (auto En : enumerate(ReductionInfos)) { in emitGlobalToListCopyFunction()
3124 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitGlobalToListCopyFunction()
3179 ArrayRef<ReductionInfo> ReductionInfos, Function *ReduceFn, in emitGlobalToListReduceFunction() argument
3212 ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in emitGlobalToListReduceFunction()
3239 for (auto En : enumerate(ReductionInfos)) { in emitGlobalToListReduceFunction()
3268 StringRef ReducerName, ArrayRef<ReductionInfo> ReductionInfos, in createReductionFunction() argument
3305 Type *RedArrayTy = ArrayType::get(Builder.getPtrTy(), ReductionInfos.size()); in createReductionFunction()
3309 for (auto En : enumerate(ReductionInfos)) { in createReductionFunction()
3341 for (auto En : enumerate(ReductionInfos)) { in createReductionFunction()
3367 checkReductionInfos(ArrayRef<OpenMPIRBuilder::ReductionInfo> ReductionInfos, in checkReductionInfos() argument
3369 for (const OpenMPIRBuilder::ReductionInfo &RI : ReductionInfos) { in checkReductionInfos()
3388 InsertPointTy CodeGenIP, ArrayRef<ReductionInfo> ReductionInfos, in createReductionsGPU() argument
3395 checkReductionInfos(ReductionInfos, /*IsGPU*/ true); in createReductionsGPU()
3405 if (ReductionInfos.size() == 0) in createReductionsGPU()
3420 ReductionInfos, ReductionGenCBKind, FuncAttrs); in createReductionsGPU()
3442 auto Size = ReductionInfos.size(); in createReductionsGPU()
3454 for (auto En : enumerate(ReductionInfos)) { in createReductionsGPU()
3465 emitShuffleAndReduceFunction(ReductionInfos, ReductionFunc, FuncAttrs); in createReductionsGPU()
3466 Function *WcFunc = emitInterWarpCopyFunction(Loc, ReductionInfos, FuncAttrs); in createReductionsGPU()
3473 for (auto En : enumerate(ReductionInfos)) { in createReductionsGPU()
3480 Builder.getInt64(MaxDataSize * ReductionInfos.size()); in createReductionsGPU()
3497 ReductionInfos, ReductionsBufferTy, FuncAttrs); in createReductionsGPU()
3499 ReductionInfos, ReductionFunc, ReductionsBufferTy, FuncAttrs); in createReductionsGPU()
3501 ReductionInfos, ReductionsBufferTy, FuncAttrs); in createReductionsGPU()
3503 ReductionInfos, ReductionFunc, ReductionsBufferTy, FuncAttrs); in createReductionsGPU()
3539 for (auto En : enumerate(ReductionInfos)) { in createReductionsGPU()
3583 ArrayRef<ReductionInfo> ReductionInfos, in createReductions() argument
3585 assert(ReductionInfos.size() == IsByRef.size()); in createReductions()
3586 for (const ReductionInfo &RI : ReductionInfos) { in createReductions()
3608 unsigned NumReductions = ReductionInfos.size(); in createReductions()
3615 for (auto En : enumerate(ReductionInfos)) { in createReductions()
3629 bool CanGenerateAtomic = all_of(ReductionInfos, [](const ReductionInfo &RI) { in createReductions()
3668 for (auto En : enumerate(ReductionInfos)) { in createReductions()
3706 for (const ReductionInfo &RI : ReductionInfos) { in createReductions()
3726 for (auto En : enumerate(ReductionInfos)) { in createReductions()