mlir-hlo/lib/Dialect/mhlo
Alexander Belyaev d4f2c767d3 [HLO] Fix HLO DynamicBroadcastInDimOp -> LHLO lowering.
The conversion had a bug in computation of strides and sizes args for std.memref_reinterpret_cast. The previous version also relied on linalg::ReshapeOp to do broadcasting when the rank of the output was higher than the rank of the input. Now the broadcasting is entirely done via descriptor modification and linalg::ReshapeOp was replaced with CopyOp.

PiperOrigin-RevId: 341379871
2020-11-09 04:24:40 -08:00
..
IR Update GetDimensionSize and SetDimensionSize ops to use I64 attribute for dimension 2020-11-06 18:03:04 -08:00
transforms [HLO] Fix HLO DynamicBroadcastInDimOp -> LHLO lowering. 2020-11-09 04:24:40 -08:00
CMakeLists.txt Add CMake files and lit configurations, enough for `ninja check-mlir-hlo` to pass on all the tests 2020-08-07 22:14:34 -07:00