colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
48 Commits 1 Branch 0 Tags 2.6 MiB
C++ 59.5%
MLIR 37%
Starlark 2.1%
CMake 0.9%
C 0.2%
Other 0.2%
Go to file
Smit Hinsu 1c535f1718 Restrict GetDimensionSize HLO op result type to 32 bit integer 
XLA implementation has this limitation and always uses 32 bit result for this instruction. This will cause mismatch between the result type in MLIR and XLA at the time of export.

This should be resolved once we have a special dialect mapping directly to HLOInstructionProto. Another option until then could be to introduce a pass to legalize mhlo itself to match XLA semantics.

PiperOrigin-RevId: 324286936
 		2020-08-03 19:30:15 -07:00
include/mlir-hlo Restrict GetDimensionSize HLO op result type to 32 bit integer 2020-08-03 19:30:15 -07:00
lib Add optional static registration for mhlo/lmhlo passes. 2020-08-03 19:29:40 -07:00
tests Constrain mhlo.const to static shaped tensors. 2020-08-03 19:29:51 -07:00
tools/mlir-hlo-opt More cleanup in mlir-hlo to prepare for the standalone build 2020-08-03 19:28:00 -07:00