Searched defs:GpuAllReduceRewriter (Results 1 – 1 of 1) sorted by relevance
30 struct GpuAllReduceRewriter { struct31 using AccumulatorFactory = std::function<Value(Value, Value)>;33 GpuAllReduceRewriter(gpu::GPUFuncOp funcOp, gpu::AllReduceOp reduceOp, in GpuAllReduceRewriter() function67 void rewrite() { in rewrite()149 T create(Args... args) { in create()155 Value getDimOp(gpu::Dimension dimension) { in getDimOp()161 Value createWorkgroupBuffer() { in createWorkgroupBuffer()172 AccumulatorFactory getFactory() { in getFactory()185 AccumulatorFactory getFactory(Region &body) { in getFactory()218 AccumulatorFactory getFactory(gpu::AllReduceOperation opName) { in getFactory()[all …]