mlir-hlo/lib/Dialect/mhlo/IR
Rahul Joshi 9902e6ee32 [HLO] Add LMHLO CollectivePermute verification.
- Extract verification of source target pairs attached to collective permute into a common
  helper function and use that to verify both MHLO and LMHLO variants.
- Change MlirGpuTestBase::ParseMlirModule to allow returning back a failure, and use
  that to update the mlir_gpu_compile_test to check the new behavior.

PiperOrigin-RevId: 362156962
2021-03-10 15:37:12 -08: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 [HLO] Add LMHLO CollectivePermute verification. 2021-03-10 15:37:12 -08: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 Canonicalize dynamic_broadcast_in_dim to own shape with rank narrowing on the shape to a corresponding tensor.cast. 2021-03-10 05:43:54 -08: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 [HLO] Add LMHLO CollectivePermute verification. 2021-03-10 15:37:12 -08: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