d166b66cba
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/40925 …ad of std.store The xla_lhlo.const lowering uses std.store to store a constant to 0-d memrefs. Update it to affine.store since such an access is trivially affine (no indices). An affine.store can always be lowered to std.store. Copybara import of the project: -- 9e18ede72fbbca107177bd742921e4cbf77adc82 by Uday Bondhugula <uday@polymagelabs.com>: [MLIR] Update lhlo.const to linalg lowering to use affine.store instead of std.store The xla_lhlo.const lowering uses std.store to store a constant to 0-d memrefs. Update it to affine.store since such an access is trivially affine (no indices). An affine.store can always be lowered to std.store. COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/tensorflow/pull/40925 from polymage-labs:lhlo_to_linalg_affine_store 9e18ede72fbbca107177bd742921e4cbf77adc82 PiperOrigin-RevId: 320623152 |
||
---|---|---|
include/mlir-hlo | ||
lib | ||
tests |