colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
mlir-hlo/tests
History
Hanhan Wang 90f0d7f935 Add support for lowering mhlo.conv to Linalg on tensors. 
This pattern only works for normal convolutions. It does not work for depthwise
convolutions. The Linalg conv ops are defined with static rank, so it only
supports 1d/2d/3d cases, which are the most typical cases.

This also refactors out the same check in lmhlo.conv lowering.

PiperOrigin-RevId: 359503527
 		2021-02-25 05:59:08 -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 [mlir][hlo] Make select ready for dynamic shapes (ranked only for now) 2021-02-18 08:08:40 -08:00
chlo_legalize_to_mhlo.mlir [MLIR][CHLO] Add legalization for `chlo.polygamma` to MHLO 2021-02-17 08:33:01 -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 Lower Expm1 kernel to math.ExpM1. 2021-02-18 04:54:23 -08:00
hlo-legalize-to-linalg.mlir Add support for lowering mhlo.conv to Linalg on tensors. 2021-02-25 05:59:08 -08:00
hlo-transform-unranked.mlir [MLIR][HLO] Remove duplicate `PopulateTransformUnrankedHloPatterns` 2021-02-23 07:50:47 -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 [mhlo] Lower int->bool to a comparison with zero 2021-02-15 06:38:09 -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 [MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. 2021-02-22 08:43:00 -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 Support dynamic-shaped operand in verification of BroadcastInDim. 2021-02-17 16:18:09 -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