colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[XLA/GPU] Migrate nested reduce emitter to take LMHLO. 

PiperOrigin-RevId: 343373088
This commit is contained in:
Tim Shen 2020-11-19 14:41:02 -08:00 committed by TensorFlow MLIR Team
parent 88642f58d3
commit 303181924c
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
include/mlir-hlo/Dialect/mhlo/IR/lhlo_ops.td
View File

@ -197,10 +197,11 @@ def LHLO_XorOp : LHLO_BinaryElementwiseOp<"xor", LHLO_PredOrIntBuffer>, BASE_HLO
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// TODO(b/139813999): specify required function signature in a type-safe way. // TODO(b/139813999): specify required function signature in a type-safe way.
def LHLO_ReduceOp: LHLO_Op<"reduce", [ //
SameVariadicOperandSize, // The region `body` may return lmhlo.TerminatorOp or mhlo.ReturnOp. We are
SingleBlockImplicitTerminator<"TerminatorOp"> // moving towards mhlo.ReturnOp, but some code that needs cleanup still assumes lmhlo.TerminatorOp.
]>, BASE_HLO_ReduceOp { // TODO(timshen): cleanup lmhlo.TerminatorOp.
def LHLO_ReduceOp: LHLO_Op<"reduce", [SameVariadicOperandSize]>, BASE_HLO_ReduceOp {
let arguments = (ins let arguments = (ins
Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$operands, Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$operands,
Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$init_values, Arg<Variadic<LHLO_Buffer>, "", [MemRead]>:$init_values,