colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
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.mlir Fold xla iota across a 1-length dimension into a zero value 2020-07-30 22:33:43 +00:00
chlo_infer_shape_type_methods.mlir 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_broadcasts.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
concatenate.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
convert.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
hlo-legalize-to-lhlo-unranked.mlir [MLIR][LHLO] Convert mhlo.dynamic_reshape -> lhlo.reshape_memref_cast. 2020-07-30 22:33:02 +00:00
hlo-legalize-to-lhlo.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
hlo-legalize-to-linalg.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
inlining.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
legalize-control-flow.mlir 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-std.mlir 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.mlir 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.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo-fuse-linalg.mlir Move HLO tests to `third_party/tensorflow/compiler/mlir/hlo/` 2020-07-30 22:32:45 +00:00
lhlo-legalize-select-and-scatter.mlir 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-affine.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo-legalize-to-gpu.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo-legalize-to-linalg.mlir PR #40925: [MLIR] Update lhlo.const to linalg lowering to use affine.store inste… 2020-07-30 22:33:51 +00:00
lhlo-legalize-to-llvm.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo-legalize-to-parallel-loops.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lhlo_ops.mlir Rename xla_lhlo dialect into lmhlo 2020-07-30 22:33:11 +00:00
lower-complex.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
lower-general-dot.mlir 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.mlir 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.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
mhlo-transform-unranked.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
ops.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
reduce.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
reshape.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
reverse.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
sink-constants-to-control-flow.mlir Cleanup build rule names in compiler/mlir/hlo to remove the redundant/obsolete xla_ prefix 2020-07-30 22:33:29 +00:00
transpose.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
tuple.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
unfuse_batch_norm.mlir Integrate LLVM at https://github.com/llvm/llvm-project/commit/305b500eaf8c 2020-07-30 22:33:38 +00:00