Skip to content

GH-90: Add workflow files for test CI jobs#4

Merged
charleshofer merged 1 commit intomasterfrom
gh90-port-rocm-gpu-ci
May 15, 2025
Merged

GH-90: Add workflow files for test CI jobs#4
charleshofer merged 1 commit intomasterfrom
gh90-port-rocm-gpu-ci

Conversation

@charleshofer
Copy link
Collaborator

Add workflow files for building and running nightly tests and tests on PRs

Story: https://github.com/ROCm/jax-internal/issues/90

@charleshofer charleshofer changed the title Add workflow files for test CI jobs GH-90: Add workflow files for test CI jobs May 13, 2025
@charleshofer charleshofer force-pushed the gh90-port-rocm-gpu-ci branch from 5111ae6 to 88ed06c Compare May 14, 2025 21:17
@charleshofer charleshofer marked this pull request as ready for review May 14, 2025 21:31
@charleshofer charleshofer force-pushed the gh90-port-rocm-gpu-ci branch from 88ed06c to d4bc4fc Compare May 14, 2025 21:36
Copy link
Contributor

@mrodden mrodden left a comment

Choose a reason for hiding this comment

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

It looks too neat.

Copy link
Contributor

@mrodden mrodden left a comment

Choose a reason for hiding this comment

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

Actually had a question I thought of later.

Comment on lines +44 to +48
- name: Checkout JAX repo
uses: actions/chekcout@v4
with:
repository: jax-ml/jax
path: jax
Copy link
Contributor

Choose a reason for hiding this comment

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

What version of jax does it checkout? rocm-main?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

Tip of upstream main. We need to hammer out how we handle tests though. It'd be nice if we were just able to submit broken tests (where it's the test case's fault) to upstream if they merge them quickly. And we talked about doing a skip list somewhere, but I haven't actually made a go at implementing that yet. One option is to just put stuff in the skip list until upstream fixes it. We could also rely on rocm-main in rocm/jax, but then we need to keep up with our sync stuff.

@charleshofer charleshofer merged commit 67c0d0f into master May 15, 2025
@charleshofer charleshofer deleted the gh90-port-rocm-gpu-ci branch May 15, 2025 15:25
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.

2 participants