colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
History
Adrian Kuegel 824bc9c425 Improve broadcast transformation to treat dynamic shapes with 1 element as scalar. 
A shape that contains exactly one element is effectively a scalar. This leads
to a speedup in cases where we have a binary op with one operand that is
effectively a scalar, because we can use the fast path.

PiperOrigin-RevId: 357515552
 		2021-02-14 23:25:41 -08:00
..
end2end Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -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
canonicalize.mlir Add MLIR generated kernel for Angle kernel. 2021-02-02 00:47:20 -08:00
chlo_infer_shape_type_methods.mlir [MLIR][KernelGen] Implement InferShapedTypeOpInterface for `mhlo.compare/select` 2020-09-17 07:10:10 -07:00
chlo_legalize_to_hlo_broadcasts.mlir Implement lowering of chlo::zeta to mhlo dialect. 2021-02-03 07:50:05 -08:00
chlo_legalize_to_mhlo.mlir Implement lowering of chlo::zeta to mhlo dialect. 2021-02-03 07:50:05 -08:00
concatenate.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
convert.mlir Fix constant folding of mhlo.convert op with i1 element types 2020-11-23 14:18:28 -08:00
hlo-legalize-gather-to-torch-index-select.mlir Add a transform for Gathers to torch_index_select. 2020-07-30 22:34:32 +00:00
hlo-legalize-to-lhlo-unranked.mlir [KERNEL_GEN] Add a pattern to bufferize `mhlo.reshape(<unranked_tensor>)`. 2021-02-10 06:32:21 -08:00
hlo-legalize-to-lhlo.mlir Integrate LLVM at llvm/llvm-project@8f7f2c4211 2021-02-02 01:54:32 -08:00
hlo-legalize-to-linalg.mlir Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
hlo-transform-unranked.mlir Improve broadcast transformation to treat dynamic shapes with 1 element as scalar. 2021-02-14 23:25:41 -08:00
inlining.mlir Rename `xla_hlo` dialect to `mhlo` 2020-07-30 22:32:50 +00:00
legalize-control-flow.mlir Integrate LLVM at llvm/llvm-project@0cf7e4b252 2020-12-16 20:30:17 -08:00
legalize-to-std.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08:00
legalize-trigonometric-to-approximation.mlir Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
legalize_to_scf.mlir Integrate LLVM at llvm/llvm-project@0cf7e4b252 2020-12-16 20:30:17 -08:00
lhlo-fuse-linalg.mlir Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
lhlo-legalize-select-and-scatter.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08:00
lhlo-legalize-to-affine.mlir [mlir][hlo] Make min/max always propagate NaNs 2021-01-25 09:04:16 -08:00
lhlo-legalize-to-gpu.mlir Integrate LLVM at llvm/llvm-project@91e7a17133 2021-02-04 13:42:31 -08:00
lhlo-legalize-to-linalg.mlir Integrate LLVM at llvm/llvm-project@2bfe27da17 2021-02-12 08:32:03 -08:00
lhlo-legalize-to-parallel-loops.mlir Integrate LLVM at llvm/llvm-project@e2d7d3cb0e 2021-01-14 18:05:33 -08:00
lhlo_gpu_ops.mlir [XLA:GPU] Add layout attributes to LHLO_GPU Convolution operations. 2020-12-17 09:26:28 -08:00
lhlo_ops.mlir [XLA/GPU] Plumb through Bitcast op for LMHLO. 2021-02-10 19:45:40 -08:00
lit.cfg.py 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
lit.site.cfg.py.in Add license header to lit.site.cfg.py.in 2020-08-07 22:22:24 -07:00
lower-complex.mlir Removed Op(Complex, Real) lowering to address complex type inference issue 2020-11-06 15:24:23 -08: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_infer_shape_type_methods.mlir Integrate LLVM at llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00
ops.mlir Add layout to mhlo::InfeedOp td. 2021-02-08 09:48:14 -08:00
optimize-hlo.mlir Add an optimization that converts some Gathers to Slices. 2020-07-30 22:34:10 +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 Sink standard dialect constants in sink_constants_to_control_flow pass 2020-08-03 19:30:29 -07: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 llvm/llvm-project@96ef4f307d 2021-01-20 07:09:47 -08:00