colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
328 Commits 1 Branch 0 Tags 2.6 MiB
Commit Graph

14 Commits

Author SHA1 Message Date
Richard Uhler 82031b356c Improve error message for improperly shaped slice indices. 
The slice indices must be rank-1 and have the same number of elements of the
rank of the operand. Give reasonable error messages for violations of these
requirements instead of a misleading error message that the types of the
indices don't all match.

PiperOrigin-RevId: 340660822
 2020-11-04 09:10:51 -08:00
Richard Uhler f9843fabe1 Use InferTypeOpInterface for HLO_SliceOp. 
Instead of having a custom builder to construct a slice op without an explicit
return type.

PiperOrigin-RevId: 339058864
 2020-10-26 09:54:13 -07:00
Jacques Pienaar 27968619b7 Verify non-scalar inputs for HLO concat 
XLA HLO concat does not accept scalars, so fail verification if this occurs. Avoids segfault when accessing an empty output shape.

PiperOrigin-RevId: 337618167
 2020-10-16 19:39:31 -07:00
Tim Shen c708bfd6d0 [MLIR] Add cbrt, reduce-precision, and bitcast ops to MHLO. 
PiperOrigin-RevId: 335109804
 2020-10-02 15:13:18 -07:00
Robert Suderman 26ac5baae4 Make mhlo.sort return variadic results instead of a tuple 
Tuple is only used on XLA's sort to return multiple inputs. MLIR supports
multiple inputs, switch to a tuple return.

PiperOrigin-RevId: 334226937
 2020-09-28 13:32:23 -07:00
Rahul Joshi 87d9e6951e [MLIR] Use MLIR provided 0DTensorOf/1DTensorOf instead of custom constraints. 
- Use MLIR provided constraints for HLO_ScalarIntTensor and HLO_DimensionTensor.
- Update unit tests to expect new error messages.

PiperOrigin-RevId: 333313131
 2020-09-23 09:57:12 -07:00
Prakalp Srivastava b7248424ae Fix return type of replica-id to unsigned int 32 tensor. 
PiperOrigin-RevId: 330012181
 2020-09-03 16:08:38 -07:00
Andy Ly 53fdda7f3e Update mhlo.constant to use a custom assembly format instead of a custom printer and parser (NFC). 
PiperOrigin-RevId: 325560779
 2020-08-07 22:23:06 -07:00
Lucy Fox d742477c02 Verify that MHLO DynamicUpdateSlice start indices have matching element types. 
HLO requires that the element types match for all start index parameters. Right now we don't catch this invalid case until export, so adding a check in the verifier so that we catch this sooner.

This also requires a small tweak to the TF InplaceUpdate op lowering.

PiperOrigin-RevId: 325463796
 2020-08-07 22:21:40 -07:00
Lucy Fox cd22ecd136 Relax DynamicBroadcastInDim verifier when dimensions are dynamic. 
For input and output dimensions which must match, we shouldn't fail in the case where one dim is dynamic and the other is static. This is insufficient information to conclude a dimension mismatch.

PiperOrigin-RevId: 325344738
 2020-08-07 22:18:38 -07:00
Andy Ly 4c8fead3e0 Add support for token operands to mhlo.tuple. 
mhlo.get_tuple_element supports extracting a mhlo.token type from a tuple. This updates the creation of tuples to allow for mhlo.token typed operands.

PiperOrigin-RevId: 324628663
 2020-08-03 19:31:15 -07:00
Stephan Herhut 9cbe5f2285 Constrain mhlo.const to static shaped tensors. 
Constants of unknown shape cannot be materialized. In most cases, one likely wants to use a scalar constant and rely on broadcasting instead.

PiperOrigin-RevId: 324252475
 2020-08-03 19:29:51 -07:00
Mehdi Amini 8900222fed Rename `xla_hlo` dialect to `mhlo` 
This is part of the current refactoring of the HLO related dialect.
`xla_hlo` will be reintroduced in a new form later.

PiperOrigin-RevId: 319916753
 2020-07-30 22:32:50 +00:00
Mehdi Amini fa057cc0bc Move HLO tests to `third_party/tensorflow/compiler/mlir/hlo/` 
Also add a localized `mlir-hlo-opt` binary for the testing of
tensorflow/compiler/mlir/hlo/... ; this directory is intended to be self-contained
and depend only on MLIR.

PiperOrigin-RevId: 319878984
 2020-07-30 22:32:45 +00:00