Searched defs:GpuAllReduceRewriter (Results 1 – 1 of 1) sorted by relevance
26 struct GpuAllReduceRewriter { struct27 using AccumulatorFactory = std::function<Value(Value, Value)>;29 GpuAllReduceRewriter(gpu::GPUFuncOp funcOp_, gpu::AllReduceOp reduceOp_, in GpuAllReduceRewriter() function63 void rewrite() { in rewrite()142 template <typename T, typename... Args> T create(Args... args) { in create()147 template <typename T> Value getDimOp(StringRef dimension) { in getDimOp()153 Value createWorkgroupBuffer() { in createWorkgroupBuffer()164 AccumulatorFactory getFactory() { in getFactory()177 AccumulatorFactory getFactory(Region &body) { in getFactory()210 AccumulatorFactory getFactory(StringRef opName) { in getFactory()[all …]