mlir-hlo

Commit Graph

Author	SHA1	Message	Date
A. Unique TensorFlower	c62fd89663	[MLIR][HLO] Add equal shapes case to rank specialization Also restructure lowering implementation to facilitate the addition or removal of special cases. PiperOrigin-RevId: 374626365	2021-05-19 05:38:42 -07:00
A. Unique TensorFlower	6af3d2df91	[MLIR][HLO] Add rank specialization with multiple non-scalar operands Add lowering pattern for rank specialization clusters with more than one non-scalar operand. The lowering resembles that of the `TransformUnrankedHlo` pass and switches cases for maximal ranks from 1 through 8. PiperOrigin-RevId: 374377002	2021-05-18 03:02:45 -07:00
A. Unique TensorFlower	474e419729	[MLIR][HLO] Generalize rank specialization with single operand The pattern can be generalized to also rank specialize operations with a single non-scalar operand. Also extract helper functions that can be reused in following specializations. PiperOrigin-RevId: 374198381	2021-05-17 08:12:55 -07:00
A. Unique TensorFlower	c514c73390	[MLIR][HLO] Extend rank specialization clustering pass Also cluster operations that operate on same shape operands. These implicitly satisfy the broadcasting semantics requirement. Also, add test cases for some cases that appear in the current MLIR-generated kernels. PiperOrigin-RevId: 374191950	2021-05-17 07:31:36 -07:00
A. Unique TensorFlower	ccd70d5717	[MLIR][HLO] Add `rank-specialization-to-scf` pass Currently the lowering is only implemented for the unary case. The n-ary case will follow. PiperOrigin-RevId: 374162772	2021-05-17 03:56:23 -07:00
A. Unique TensorFlower	313d24bc8f	[MLIR][HLO] Add `rank-specialization-cluster` pass Add a pass to cluster unranked C/HLO operations in one `chlo.rank_specialization_cluster` op. The C/HLO operations are moved to the body of the operation. Later passes can use this to rank-specialize all these operations together. PiperOrigin-RevId: 373336725	2021-05-12 03:46:01 -07:00

6 Commits