Skip to content

Actions: jax-ml/jax

Actions

ROCm GPU Post-Merge Check

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
rocm-ci.yml
215 workflow runs
215 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[sharding_in_types] Add sharding rules for the following primitives: ROCm GPU Post-Merge Check #215: Commit d695aa4 pushed by copybara-service bot
February 22, 2025 18:46 10m 4s main
main
February 22, 2025 18:46 10m 4s
Update XLA dependency to use revision ROCm GPU Post-Merge Check #214: Commit a75bea5 pushed by copybara-service bot
February 22, 2025 10:30 9m 21s main
main
February 22, 2025 10:30 9m 21s
[sharding_in_types] Allow auto_axes and explicit_axes to take num… ROCm GPU Post-Merge Check #213: Commit 7c4fe2a pushed by copybara-service bot
February 22, 2025 02:49 10m 13s main
main
February 22, 2025 02:49 10m 13s
Reverts 6e83de5909b5dad6827c1682726c840c9688b32d ROCm GPU Post-Merge Check #212: Commit 3407785 pushed by copybara-service bot
February 22, 2025 02:34 9m 19s main
main
February 22, 2025 02:34 9m 19s
[sharding_in_types] Make slice and ellipsis work with `.at[...].get(o… ROCm GPU Post-Merge Check #211: Commit 80f18de pushed by copybara-service bot
February 22, 2025 02:25 9m 19s main
main
February 22, 2025 02:25 9m 19s
Allow casting to the same axis type ROCm GPU Post-Merge Check #210: Commit 629426f pushed by copybara-service bot
February 21, 2025 22:53 11m 2s main
main
February 21, 2025 22:53 11m 2s
Implement Jax CPU/GPU callbacks with XLA's FFI. ROCm GPU Post-Merge Check #209: Commit 6e83de5 pushed by copybara-service bot
February 21, 2025 21:34 9m 21s main
main
February 21, 2025 21:34 9m 21s
roofline: Handle ClosedJaxpr instances. ROCm GPU Post-Merge Check #208: Commit b3fcba7 pushed by copybara-service bot
February 21, 2025 21:19 9m 5s main
main
February 21, 2025 21:19 9m 5s
[JAX] Implement an initial object API for colocated Python ROCm GPU Post-Merge Check #207: Commit 96b7dba pushed by copybara-service bot
February 21, 2025 20:58 9m 8s main
main
February 21, 2025 20:58 9m 8s
Reverts 655267609b5589fda8358ae7aaf2eb832036407a ROCm GPU Post-Merge Check #206: Commit 6c83d43 pushed by copybara-service bot
February 21, 2025 19:59 9m 11s main
main
February 21, 2025 19:59 9m 11s
Update several remaining lax.linalg primitives to use registration he… ROCm GPU Post-Merge Check #205: Commit c4418c1 pushed by copybara-service bot
February 21, 2025 18:18 13m 54s main
main
February 21, 2025 18:18 13m 54s
Use new ML Build CUDA images ROCm GPU Post-Merge Check #204: Commit 5089fb0 pushed by copybara-service bot
February 21, 2025 18:16 2m 14s main
main
February 21, 2025 18:16 2m 14s
Update Windows docker image to pin it to a sha ROCm GPU Post-Merge Check #203: Commit a073639 pushed by copybara-service bot
February 21, 2025 18:14 9m 16s main
main
February 21, 2025 18:14 9m 16s
Refactor Jax FFI lowering to prepare for implementing CPU/GPU callbac… ROCm GPU Post-Merge Check #202: Commit 2d1bc5c pushed by copybara-service bot
February 21, 2025 17:46 9m 13s main
main
February 21, 2025 17:46 9m 13s
Don't set PYTHONWARNINGS=error for tests that use TensorFlow. ROCm GPU Post-Merge Check #201: Commit 673a02d pushed by copybara-service bot
February 21, 2025 17:23 11m 58s main
main
February 21, 2025 17:23 11m 58s
Update lax.linalg.qr primitives to use registration helper functions. ROCm GPU Post-Merge Check #200: Commit ed10003 pushed by copybara-service bot
February 21, 2025 17:15 10m 50s main
main
February 21, 2025 17:15 10m 50s
Disable tests/debug_info_test.py:test_vjp_of_jit test. Currently fail… ROCm GPU Post-Merge Check #199: Commit 87a7158 pushed by copybara-service bot
February 21, 2025 16:50 9m 13s main
main
February 21, 2025 16:50 9m 13s
Set the mesh of the sharding during broadcast in vmap so that we don'… ROCm GPU Post-Merge Check #198: Commit 66037d1 pushed by copybara-service bot
February 21, 2025 16:05 8m 45s main
main
February 21, 2025 16:05 8m 45s
[CI] Enable workflow_dispatch for the continuous workflow ROCm GPU Post-Merge Check #197: Commit c664a0c pushed by copybara-service bot
February 21, 2025 14:59 9m 12s main
main
February 21, 2025 14:59 9m 12s
Merge pull request #26657 from gnecula:debug_info_test ROCm GPU Post-Merge Check #196: Commit cb8ff82 pushed by copybara-service bot
February 21, 2025 14:21 9m 12s main
main
February 21, 2025 14:21 9m 12s
Merge pull request #26649 from jakevdp:wrapped-fun ROCm GPU Post-Merge Check #195: Commit 237b794 pushed by copybara-service bot
February 21, 2025 13:58 9m 10s main
main
February 21, 2025 13:58 9m 10s
Update internal unop primitive helper to pass kwargs to dtype rule. ROCm GPU Post-Merge Check #194: Commit 09325d9 pushed by copybara-service bot
February 21, 2025 12:52 16m 5s main
main
February 21, 2025 12:52 16m 5s
Update lax.linalg.lu primitive to use registration helper functions. ROCm GPU Post-Merge Check #193: Commit 126909b pushed by copybara-service bot
February 21, 2025 12:50 9m 6s main
main
February 21, 2025 12:50 9m 6s
Start adding primitive registration helper functions to lax.linalg. ROCm GPU Post-Merge Check #192: Commit a981e1c pushed by copybara-service bot
February 21, 2025 12:05 9m 20s main
main
February 21, 2025 12:05 9m 20s
Reverts 9ad9c3d3c2b6b5c3ea736af9b4d8c595537de93c ROCm GPU Post-Merge Check #191: Commit 6552676 pushed by copybara-service bot
February 21, 2025 11:30 9m 27s main
main
February 21, 2025 11:30 9m 27s