mlir-hlo

Commit Graph

Author	SHA1	Message	Date
Rahul Joshi	5adb7c6e12	[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations. - XLA:HLO -> LMHLO conversion drops all token arguments and return values, however custom calls that users write still expect to get buffer pointers for these token types. - To be able to support this, add an optional call target argument mapping attribute to LMHLO custom calls. When this attribute is present, it indicates the number of arguments and returns that the custom call expects and also indicates which LMHLO arg() or output() maps to which arg or result number of the custom call. PiperOrigin-RevId: 358826664	2021-02-22 08:43:00 -08:00
A. Unique TensorFlower	51cd4200b6	Make LMHLO's Dot have the same power as MHLO's DotGeneral. PiperOrigin-RevId: 337391565	2020-10-15 15:09:06 -07:00

Author

SHA1

Message

Date

Rahul Joshi

5adb7c6e12

[MLIR:LHLO] Add optional call target arg mapping to LMHLO CustomCall operations.

- XLA:HLO -> LMHLO conversion drops all token arguments and return values, however
  custom calls that users write still expect to get buffer pointers for these token types.
- To be able to support this, add an optional call target argument mapping attribute to
  LMHLO custom calls. When this attribute is present, it indicates the number of
  arguments and returns that the custom call expects and also indicates which LMHLO
  arg() or output() maps to which arg or result number of the custom call.

PiperOrigin-RevId: 358826664

2021-02-22 08:43:00 -08:00

A. Unique TensorFlower

51cd4200b6

Make LMHLO's Dot have the same power as MHLO's DotGeneral.

PiperOrigin-RevId: 337391565

2020-10-15 15:09:06 -07:00

2 Commits