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

[MLIR][KernelGen] Enable cluster rank specialization 

Replace the previously used `TransformUnrankedHloPass` which rank-specializes
only one operation at a time. The new generalized rank specialization clusters
compatible operations and rank-specializes them collectively.

PiperOrigin-RevId: 376127752
This commit is contained in:
A. Unique TensorFlower 2021-05-27 02:43:43 -07:00 committed by TensorFlow MLIR Team
parent a4fa6afa07
commit b32f885ad7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1029,7 +1029,7 @@ def HLO_TupleOp : HLO_Op<"tuple", [NoSideEffect]> {
} }
def HLO_CompareOp: HLO_Op<"compare", [NoSideEffect, SameTypeOperands, def HLO_CompareOp: HLO_Op<"compare", [NoSideEffect, SameTypeOperands,
SameOperandsAndResultShape, SameOperandsAndResultShape, Elementwise,
DeclareOpInterfaceMethods<InferShapedTypeOpInterface, DeclareOpInterfaceMethods<InferShapedTypeOpInterface,
["inferReturnTypeComponents", "reifyReturnTypeShapes"]>]> { ["inferReturnTypeComponents", "reifyReturnTypeShapes"]>]> {
let summary = "Comparison operator"; let summary = "Comparison operator";