Rahul Joshi 8e3890e8e8 [MLIR:HLO] Add AllGather and AllToAll operations to LMHLO dialect. ... - Use a common base class to for AllReduce, AllGather, and AllToAll in the ODS spec. - Add basic verification for replica groups attribute. PiperOrigin-RevId: 354969654		2021-02-01 10:23:46 -08:00
..
CMakeLists.txt	[MLIR:HLO] Generate enum decls for HLO and LHLO GPU dialects.	2020-12-02 11:39:23 -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 zeta and broadcasting_zeta to chlo dialect.	2021-01-29 03:22:52 -08:00
hlo_ops.cc	[MLIR][HLO] Implement type inference for `is_finite` op	2021-01-28 00:56: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_patterns.td	Add canonicalizer for Reshape(Broadcast(X)) pattern when it is an identity sequence	2020-11-19 02:32:45 -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	[MLIR:HLO] Add AllGather and AllToAll operations to LMHLO dialect.	2021-02-01 10:23:46 -08:00
lhlo_ops_structs.cc	Make LMHLO's Dot have the same power as MHLO's DotGeneral.	2020-10-15 15:09:06 -07:00
mhlo_canonicalize.td	[KERNEL_GEN] Restrict broadcast -> reshape canonicalization to identity dims.	2021-01-07 15:30:28 -08:00