ci: Add OneAPI artifact build script and auditwheel support#38170
ci: Add OneAPI artifact build script and auditwheel support#38170ashahba wants to merge 1 commit into
Conversation
There was a problem hiding this comment.
Code Review
This pull request introduces support for building JAX OneAPI artifacts (jax-oneapi-plugin and jax-oneapi-pjrt) by adding a new build script, defining the default OneAPI version in the environment configuration, and updating the auditwheel utility to verify the compliance of the new wheels. The review feedback suggests validating the artifact parameter early in the build script to avoid a deeply nested conditional block, and using Bash's built-in parameter expansion instead of spawning a subshell for basename in the auditwheel script.
Important
The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.
- ci/build_oneapi_artifacts.sh: new script to build jax-oneapi-pjrt and jax-oneapi-plugin wheels, following the same pattern as the existing build_artifacts.sh - ci/utilities/run_auditwheel.sh: add oneapi wheel patterns for manylinux compliance verification Signed-off-by: Abolfazl Shahbazi <12436063+ashahba@users.noreply.github.com>
6ba0705 to
1d3c7df
Compare
|
All comments Gemini bot review feedback were addressed. |
Summary
This PR adds CI infrastructure to build Intel GPU (oneAPI) JAX wheels and verify their manylinux compliance.
Test Plan: