Status: Approved
Initial version: 03/12/2024
Last updated: 03/15/2024
Discussion thread: GitHub
Several features have been added to MHLO in the past year, which frameworks want
to leverage and members of the community have made requests for them as well.
This includes: MHLO
custom_call
backend_config to take a DictionaryAttr.
The current StableHLO custom_call op does not support this feature. There are several
occurrences of users working around this gap in custom_call today, examples -
JAX uses unregistered attributes
to hold the DictionaryAttr or serializing the dictionary as a string
to pass around for the StableHLO custom_call op.
We propose standardizing the StableHLO custom_call op to extend backend_config
to take a DictionaryAttr so they can be used safely by the community without
workarounds. This will help to unify metadata under a single DictionaryAttr which
provides more stable serialization of custom_call metadata, a feature that is
desired by frameworks and compilers. Standardizing this feature to StableHLO
will benefit the entire ecosystem.
Note: backend_config will continue to accept strings as before.
Open tickets for this request: #637, #741
Please refer spec.md changes in this PR to view the diff vs original spec.