Skip to content

Update XLA dependency to use revision http://github.com/openxla/xla/c… #15077

Update XLA dependency to use revision http://github.com/openxla/xla/c…

Update XLA dependency to use revision http://github.com/openxla/xla/c… #15077

Workflow file for this run

name: Bazel nobuild
on:
pull_request:
branches:
- main
push:
branches:
- main
- 'release/**'
permissions: {}
concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
# Don't cancel in-progress jobs for main/release branches.
cancel-in-progress: ${{ !contains(github.ref, 'release/') && github.ref != 'main' }}
jobs:
bazel-dependency-violations:
if: github.repository_owner == 'jax-ml'
# Begin Presubmit Naming Check - name modification requires internal check to be updated
name: Bazel dependency violations
# End Presubmit Naming Check github-bazel-nobuild
runs-on: linux-x86-n4-16
container: us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest # zizmor: ignore[unpinned-images]
defaults:
run:
shell: bash
steps:
- uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
with:
persist-credentials: false
- name: Configure JAX build
run: python build/build.py build --wheels=jaxlib,jax-cuda-plugin,jax-cuda-pjrt --configure_only
- name: Analyze JAX build
timeout-minutes: 60
run: |
EXCLUDED_TARGETS=(
# Targets dependent on missing TEST dependencies
-//jax/experimental/jax2tf/tests:all
-//jax/experimental/jax2tf/tests/flax_models:all
-//jax/experimental/jax2tf/tests/back_compat_testdata:all
-//benchmarks/mosaic:all
)
# Include specific root targets, but skip jax_wheel_with_internal_test_util as it fails to build
TARGETS=(
":jax_sources"
":build_wheel"
":jax_wheel"
":jax_wheel_editable"
":jax_source_package"
":wheel_additives"
":jax_py_import"
":jax_wheel_size_test"
)
bazel build \
--nobuild \
--color=yes \
--keep_going \
--config=cuda_libraries_from_stubs \
-- \
//jaxlib/... \
//jax/... \
//benchmarks/... \
//build/... \
//third_party/... \
"${TARGETS[@]}" \
"${EXCLUDED_TARGETS[@]}"