Skip to content

ci: Add OneAPI artifact build script and auditwheel support#38170

Open
ashahba wants to merge 1 commit into
jax-ml:mainfrom
Intel-tensorflow:ashahba/oneapi-ci-build
Open

ci: Add OneAPI artifact build script and auditwheel support#38170
ashahba wants to merge 1 commit into
jax-ml:mainfrom
Intel-tensorflow:ashahba/oneapi-ci-build

Conversation

@ashahba
Copy link
Copy Markdown
Contributor

@ashahba ashahba commented Jun 3, 2026

Summary

This PR adds CI infrastructure to build Intel GPU (oneAPI) JAX wheels and verify their manylinux compliance.

  • ci/build_oneapi_artifacts.sh: New script to build jax-oneapi-pjrt and jax-oneapi-plugin wheels. Follows the same structure and conventions as the existing build_artifacts.sh, supporting default, nightly, and release artifact types, RBE, and remote Bazel cache.
  • ci/utilities/run_auditwheel.sh: Add jaxoneapipjrtwhl and jaxoneapipluginwhl patterns so that manylinux compliance is verified for oneAPI wheels alongside existing CUDA and ROCm wheels.

Test Plan:

  • ./ci/build_oneapi_artifacts.sh jax-oneapi-pjrt builds successfully
  • ./ci/build_oneapi_artifacts.sh jax-oneapi-plugin builds successfully
  • auditwheel passes for the produced oneAPI wheels

Copy link
Copy Markdown
Contributor

@gemini-code-assist gemini-code-assist Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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.

Comment thread ci/build_oneapi_artifacts.sh
Comment thread ci/build_oneapi_artifacts.sh Outdated
Comment thread ci/utilities/run_auditwheel.sh Outdated
- 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>
@ashahba ashahba force-pushed the ashahba/oneapi-ci-build branch from 6ba0705 to 1d3c7df Compare June 3, 2026 21:21
@ashahba
Copy link
Copy Markdown
Contributor Author

ashahba commented Jun 3, 2026

All comments Gemini bot review feedback were addressed.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant