Skip to content

Improve ROCm pytest results handling#716

Open
psanal35 wants to merge 44 commits intoamd-mainfrom
test-ups-collect-pytest-results
Open

Improve ROCm pytest results handling#716
psanal35 wants to merge 44 commits intoamd-mainfrom
test-ups-collect-pytest-results

Conversation

@psanal35
Copy link

@psanal35 psanal35 commented Feb 25, 2026

This PR improves the ROCm test workflow by collecting and archiving pytest results.

Specifically, it:

  • Generates structured pytest result output
  • Captures run-manifest information
  • Packages test logs into a archive
  • Uploads the logs to shared S3

charleshofer and others added 30 commits February 24, 2026 09:38
…tignore (#563)

When jaxlib was built in debug more, an assertion in LLVM code that lazy-loads VHLO dialect could fire, since the code path could execute in a multi-threaded environment, and LLVM dialect repositories aren't thread safe to modify.

This patch applies the same changes that upstream makes to fix this: jax-ml@48c8762

(this includes disabling a call to `jax_mlir_ext.enter_multi_threaded_execution(context)` in `mlir.py`. Presumably, the whole functionality related to `enter_multi_threaded_execution()` multithreaded checks isn't ready yet, and it was prematurely rolled into the production code.

Manual testing
(forgot this skip in the previous PR)
Co-authored-by: Daniel Suo <danielsuo@gmail.com>
Co-authored-by: Jake VanderPlas <jakevdp@google.com>
@psanal35 psanal35 force-pushed the test-ups-collect-pytest-results branch 2 times, most recently from 7b90ff6 to d2cb24c Compare February 27, 2026 14:01
@psanal35 psanal35 force-pushed the test-ups-collect-pytest-results branch 21 times, most recently from 2694927 to fc56ad8 Compare March 2, 2026 03:37
@psanal35 psanal35 force-pushed the test-ups-collect-pytest-results branch from fc56ad8 to c58f3e9 Compare March 2, 2026 17:38
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.

10 participants