[PJRT] Add support of passing per-compilation compile options (#19438) #15789
Annotations
1 error and 1 warning
Getting failed jobs
Process completed with exit code 1.
|
Post to Discord on Failure
embed field value must be shorter than 1024, got 1071
[`9b96886`](https://github.com/iree-org/iree/commit/9b96886545b20e12df216cae3c431c7d522bf73c) [PJRT] Add support of passing per-compilation compile options (#19438)
As discussed in
https://github.com/iree-org/iree/pull/19418#discussion_r1876524772,
https://github.com/iree-org/iree/pull/19418#pullrequestreview-2490115222
and https://github.com/iree-org/iree/pull/19418#discussion_r1877191103,
here we support to read `env_option_overrides` as IREE compile flags
from `compile_options` passed by frontends like JAX in a per-compilation
basis.
Most of these code already exists but has been commented due to some
problems: `compile_options` was not yet available in that time, but it's
now introduced by #19369.
A simple use case is shown below, also as a test case:
https://github.com/iree-org/iree/blob/c37a80212dd4a541762fc9fdaaa615b6d0a62829/integrations/pjrt/test/test_compile_options.py#L9-L15
ci-exactly: build_packages, test_pjrt
---------
Signed-off-by: PragmaTwice <[email protected]>
Co-authored-by: Scott Todd <[email protected]>
|
Loading