5adb7c6e12
- 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 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| chlo_canonicalize.td | ||
| chlo_ops.cc | ||
| hlo_ops.cc | ||
| hlo_ops_base_enums.cc | ||
| hlo_ops_base_structs.cc | ||
| hlo_patterns.td | ||
| infer_fusibility_op_interface.cc | ||
| init.cc | ||
| lhlo_gpu_ops.cc | ||
| lhlo_gpu_ops_enums.cc | ||
| lhlo_gpu_ops_structs.cc | ||
| lhlo_ops.cc | ||
| lhlo_ops_structs.cc | ||
| mhlo_canonicalize.td | ||