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

[MLIR] Add conversion between XLA Fusion and MHLO FusionOp. 

PiperOrigin-RevId: 324279065
This commit is contained in:
Tim Shen 2020-07-31 13:58:01 -07:00 committed by TensorFlow MLIR Team
parent 9cbe5f2285
commit fcb91fb0b9
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td
View File

@ -40,6 +40,14 @@ class HLO_Op<string mnemonic, list<OpTrait> traits> :
let verifier = [{ return Verify(*this); }]; let verifier = [{ return Verify(*this); }];
} }
def HLO_LOOP_FUSION : StrEnumAttrCase<"kLoop">;
def HLO_INPUT_FUSION : StrEnumAttrCase<"kInput">;
def HLO_OUTPUT_FUSION : StrEnumAttrCase<"kOutput">;
def HLO_CUSTOM_FUSION : StrEnumAttrCase<"kCustom">;
def HLO_FusionKindAttr : StrEnumAttr<"FusionKind", "fusion kind", [
HLO_LOOP_FUSION, HLO_INPUT_FUSION, HLO_OUTPUT_FUSION, HLO_CUSTOM_FUSION
]>;
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
// MHLO nullary op definitions. // MHLO nullary op definitions.
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//
@ -1375,7 +1383,8 @@ def HLO_FusionOp : HLO_Op<"fusion", []> {
let regions = (region SizedRegion<1>:$fused_computation); let regions = (region SizedRegion<1>:$fused_computation);
let arguments = (ins let arguments = (ins
Variadic<HLO_TensorOrTuple>:$operands Variadic<HLO_TensorOrTuple>:$operands,
OptionalAttr<HLO_FusionKindAttr>:$fusion_kind
); );
let results = (outs let results = (outs