mlir-hlo

Commit Graph

Author	SHA1	Message	Date
Tim Shen	6fa6974e8d	[XLA/GPU] Plumb through Bitcast op for LMHLO. Also remove BitcastOp. XLA bitcast requires the input buffer to alias the output buffer, which makes bitcast always a no-op. PiperOrigin-RevId: 356884383	2021-02-10 19:45:40 -08:00
Rahul Joshi	b251712b1d	[XLA:GPU] Add conversion from HLO -> MLIR LMHLO for TriangularSolve - Also add layout attributes for inputs and output for error checking. PiperOrigin-RevId: 355863625	2021-02-05 09:18:02 -08:00
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
Rahul Joshi	44deae2aa1	[MLIR:HLO] Extend AllReduce to support multiple inputs and results (to model tuples). - Instead of SameTypeOperands, add custom verification to check if operands and results pairwise have the same type. PiperOrigin-RevId: 353986341	2021-01-26 17:25:22 -08:00
Alexander Belyaev	3d930d08c2	[HLO] Delete LHLO memref cast ops and migrate to STD ones. PiperOrigin-RevId: 340663578	2020-11-04 09:26:34 -08:00
Mehdi Amini	7c4a5d62b5	Rename xla_lhlo dialect into lmhlo Following on the plan of isolating the compiler/mlir/hlo directory. Another xla_lhlo dialect will be created under compiler/mlir/xla/ later. PiperOrigin-RevId: 320210326	2020-07-30 22:33:11 +00:00
Mehdi Amini	8900222fed	Rename `xla_hlo` dialect to `mhlo` This is part of the current refactoring of the HLO related dialect. `xla_hlo` will be reintroduced in a new form later. PiperOrigin-RevId: 319916753	2020-07-30 22:32:50 +00:00
Mehdi Amini	fa057cc0bc	Move HLO tests to `third_party/tensorflow/compiler/mlir/hlo/` Also add a localized `mlir-hlo-opt` binary for the testing of tensorflow/compiler/mlir/hlo/... ; this directory is intended to be self-contained and depend only on MLIR. PiperOrigin-RevId: 319878984	2020-07-30 22:32:45 +00:00

8 Commits