derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
55 lines
1.4 KiB
YAML
55 lines
1.4 KiB
YAML
name: Test JAX Wheels
|
|
|
|
on:
|
|
workflow_dispatch:
|
|
inputs:
|
|
test_runs_on:
|
|
required: true
|
|
type: string
|
|
default: "linux-mi325-1gpu-ossci-rocm"
|
|
image_name:
|
|
required: true
|
|
description: JAX docker image to run tests with
|
|
type: string
|
|
jax_version:
|
|
description: Version of JAX to install
|
|
required: false
|
|
type: string
|
|
jax_plugin_branch:
|
|
required: true
|
|
description: JAX plugin branch to checkout
|
|
type: string
|
|
default: "rocm-jaxlib-v0.6.0"
|
|
|
|
workflow_call:
|
|
inputs:
|
|
test_runs_on:
|
|
required: true
|
|
type: string
|
|
image_name:
|
|
required: true
|
|
description: JAX docker image to run tests with
|
|
type: string
|
|
jax_version:
|
|
description: Version of JAX to install instead of the one on the docker image
|
|
required: false
|
|
type: string
|
|
jax_plugin_branch:
|
|
description: JAX plugin branch to checkout to use for test scripts
|
|
type: string
|
|
default: "rocm-jaxlib-v0.8.0"
|
|
|
|
permissions:
|
|
contents: read
|
|
|
|
jobs:
|
|
test_wheels:
|
|
name: Test
|
|
runs-on: ${{ inputs.test_runs_on }}
|
|
steps:
|
|
- name: Checkout
|
|
uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1
|
|
with:
|
|
repo: rocm/rocm-jax
|
|
# TODO: Add steps for creating the JAX docker image with an install of TheRock and then running JAX tests on the container
|