colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/lib/Dialect/mhlo/transforms
History
Uday Bondhugula d166b66cba PR #40925: [MLIR] Update lhlo.const to linalg lowering to use affine.store inste… 
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
 		2020-07-30 22:33:51 +00:00
..
canonicalize.td Move the HLO/LHLO dialects to a new directory: tensorflow/compiler/mlir/hlo 2020-07-30 22:32:32 +00:00
chlo_legalize_to_hlo.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
chlo_legalize_to_hlo_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
hlo_legalize_to_lhlo.cc Rename XlaHloDialect class into MhloDialect following the recent dialect namespace renaming 2020-07-30 22:33:20 +00:00
legalize_control_flow.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_tanh_to_approximation.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_to_linalg.cc PR #40925: [MLIR] Update lhlo.const to linalg lowering to use affine.store inste… 2020-07-30 22:33:51 +00:00
legalize_to_standard.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
legalize_to_standard_patterns.td Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_copy_removal.cc Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo_fuse_linalg.cc Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo_legalize_to_affine.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_legalize_to_gpu.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lhlo_legalize_to_llvm.cc [MLIR][LHLO] Lower ReshapeMemRefCastOp to LLVM. 2020-07-30 22:33:47 +00:00
lhlo_legalize_to_llvm_pass.cc Add mlir_cpu_runner tests infra for CHLO->LHLO->LLVM lowering. 2020-07-30 22:33:25 +00:00
lhlo_legalize_to_parallel_loops.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lower_complex.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lower_complex_patterns.td Move XLA-independent transforms to the new MLIR-HLO directory 2020-07-30 22:32:40 +00:00
lower_general_dot.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
materialize_broadcasts.cc Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
materialize_broadcasts_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
mhlo_fusion.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
sink_constants_to_control_flow.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
test_infer_shaped_type_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
transform_unranked_hlo.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
unfuse_batch_norm.cc Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
unfuse_batch_norm_pass.cc Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00