colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/lib/Dialect/mhlo/IR
History
dfki-jugr 6bc854f5d9 PR #48667: [mlir-hlo] Added RegionBranchOpInterfaces to lmhlo operations. 
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/48667

Added RegionBranchOpInterfaces to lmhlo operations that use regions.
This is needed, since the bufferization features in MLIR have to reason about the control flow within these operations.
Copybara import of the project:

--
572fd7d850a46630b812da84e9094280f89f259e by Julian Gross <julian.gross@dfki.de>:

Added RegionBranchOpInterfaces to lmhlo operations.

PiperOrigin-RevId: 372070825
 		2021-05-05 00:27:56 -07:00
..
CMakeLists.txt [HLO] Add LMHLO CollectivePermute verification. 2021-03-10 15:37:12 -08:00
chlo_canonicalize.td Add chlo.constant_like op which splats a constant to shape of operand 2020-08-11 14:54:48 -07:00
chlo_ops.cc Add MinimumBroadcastShapesOp to chlo dialect. 2021-03-01 02:23:52 -08:00
hlo_ops.cc Introduce constant folds for ReduceOp with single LogicalAnd or LogicalOr op. 2021-04-26 15:11:27 -07:00
hlo_ops_base_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08:00
hlo_ops_base_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
hlo_ops_common.cc [HLO] Add LMHLO CollectivePermute verification. 2021-03-10 15:37:12 -08:00
hlo_patterns.td Restrict canonicalization to avoid changing type 2021-03-16 16:54:05 -07:00
infer_fusibility_op_interface.cc More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00
init.cc Add GPU specific LMHLO level ops 2020-10-14 11:23:55 -07:00
lhlo_gpu_ops.cc [mlir][NFC] Replace usages or mlir/IR/StandardTypes.h with mlir/IR/BuiltinTypes.h 2020-12-11 19:01:25 -08:00
lhlo_gpu_ops_enums.cc [MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects. 2020-12-02 11:39:23 -08:00
lhlo_gpu_ops_structs.cc Make LMHLO's Dot have the same power as MHLO's DotGeneral. 2020-10-15 15:09:06 -07:00
lhlo_ops.cc PR #48667: [mlir-hlo] Added RegionBranchOpInterfaces to lmhlo operations. 2021-05-05 00:27:56 -07:00
lhlo_ops_structs.cc [MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. 2021-02-22 08:43:00 -08:00
mhlo_canonicalize.td [KERNEL_GEN] Restrict broadcast -> reshape canonicalization to identity dims. 2021-01-07 15:30:28 -08:00