[PJRT] Add support of passing per-compilation compile options (#19438) #15789
pkgci.yml
on: push
setup
/
setup
4s
Matrix: Regression Test / test_models
Matrix: Regression Test / test_regression_suite
Matrix: Test ONNX / test_onnx_models
Matrix: Test ONNX / test_onnx_ops
Matrix: Test PJRT plugin / Build and test
Matrix: Test Sharktank / test_sharktank_models
Waiting for pending jobs
Unit Test
/
Linux (x86_64)
2m 39s
Test AMD MI250
/
test_mi250
5m 35s
Test AMD MI300
/
test_mi300
10m 51s
Test AMD W7900
/
test_w7900
1m 57s
Test Android
/
android_arm64
6m 30s
Test RISC-V 64
/
riscv64
9m 25s
Test TensorFlow
/
Linux (x86_64)
1m 17s
pkgci_summary
/
summary
2s
Annotations
4 errors and 1 warning
Test ONNX / test_onnx_ops :: cpu_llvm_sync
Process completed with exit code 1.
|
Regression Test / test_regression_suite :: amdgpu_rocm_mi300_gfx942
Process completed with exit code 1.
|
Regression Test / test_regression_suite :: amdgpu_rocm_mi250_gfx90a
Process completed with exit code 1.
|
pkgci_summary / summary
Process completed with exit code 1.
|
pkgci_summary / summary
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]>
|
Artifacts
Produced during runtime
Name | Size | |
---|---|---|
linux_x86_64_release_packages
|
74.9 MB |
|
onnx_ops_cpu_llvm_sync.json
|
2.52 KB |
|