Skip to content

Actions: jax-ml/jax

CI - Bazel CUDA tests (RBE)

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
1,851 workflow runs
1,851 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

internal: move more NumPy APIs to ensure_arraylike
CI - Bazel CUDA tests (RBE) #101: Pull request #26063 opened by jakevdp
January 23, 2025 16:27 3m 12s jakevdp:ensure-arraylike
January 23, 2025 16:27 3m 12s
[jax:custom_partitioning] Support SdyShardingRule with multiple leading
CI - Bazel CUDA tests (RBE) #98: Pull request #25886 synchronize by copybara-service bot
January 23, 2025 15:45 10m 22s test_715515514
January 23, 2025 15:45 10m 22s
Don't calculate tracing debug_info twice
CI - Bazel CUDA tests (RBE) #97: Pull request #26061 synchronize by copybara-service bot
January 23, 2025 15:37 9m 9s test_718851023
January 23, 2025 15:37 9m 9s
Remove axis_name from pmap_unmapped_aval_handlers
CI - Bazel CUDA tests (RBE) #96: Pull request #26059 synchronize by copybara-service bot
January 23, 2025 15:21 7m 59s test_718846546
January 23, 2025 15:21 7m 59s
Don't calculate tracing debug_info twice
CI - Bazel CUDA tests (RBE) #95: Pull request #26061 synchronize by copybara-service bot
January 23, 2025 15:18 10m 20s test_718851023
January 23, 2025 15:18 10m 20s
Don't calculate tracing debug_info twice
CI - Bazel CUDA tests (RBE) #94: Pull request #26061 opened by copybara-service bot
January 23, 2025 15:11 6m 26s test_718851023
January 23, 2025 15:11 6m 26s
Remove code that sets CUDA rpath option from .bazelrc.
CI - Bazel CUDA tests (RBE) #93: Pull request #26060 opened by hawkinsp
January 23, 2025 15:11 11m 58s hawkinsp:cuda
January 23, 2025 15:11 11m 58s
#sdy unskip JAX Shardy tests that are already passing
CI - Bazel CUDA tests (RBE) #92: Pull request #26055 synchronize by copybara-service bot
January 23, 2025 14:51 19m 0s test_718786627
January 23, 2025 14:51 19m 0s
Remove axis_name from pmap_unmapped_aval_handlers
CI - Bazel CUDA tests (RBE) #91: Pull request #26059 opened by copybara-service bot
January 23, 2025 14:50 15m 52s test_718846546
January 23, 2025 14:50 15m 52s
[better_errors] Port the Pallas debug info mechanisms to the new JAX debugging info
CI - Bazel CUDA tests (RBE) #90: Pull request #26058 synchronize by gnecula
January 23, 2025 14:48 9m 51s gnecula:debug_info_pallas
January 23, 2025 14:48 9m 51s
Fix typo and improve readability of workflow documentation
CI - Bazel CUDA tests (RBE) #88: Pull request #26054 synchronize by copybara-service bot
January 23, 2025 14:12 8m 49s test_718687264
January 23, 2025 14:12 8m 49s
[Mosaic GPU] Add support for tiled loads/stores with sub-byte types
CI - Bazel CUDA tests (RBE) #87: Pull request #26057 opened by copybara-service bot
January 23, 2025 13:48 10m 57s test_718283869
January 23, 2025 13:48 10m 57s
[Pallas:MGPU] Add helpers to make writing core_map kernels less verbose
CI - Bazel CUDA tests (RBE) #86: Pull request #26056 opened by copybara-service bot
January 23, 2025 13:41 13m 2s test_718824070
January 23, 2025 13:41 13m 2s
[Mosaic GPU] Remove an unnecessary restriction in the vector.store lowering
CI - Bazel CUDA tests (RBE) #85: Pull request #26035 synchronize by copybara-service bot
January 23, 2025 12:08 7m 0s test_718385442
January 23, 2025 12:08 7m 0s
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
CI - Bazel CUDA tests (RBE) #84: Pull request #26032 synchronize by copybara-service bot
January 23, 2025 11:38 5m 14s test_718302204
January 23, 2025 11:38 5m 14s
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
CI - Bazel CUDA tests (RBE) #83: Pull request #26032 synchronize by copybara-service bot
January 23, 2025 11:24 8m 56s test_718302204
January 23, 2025 11:24 8m 56s
[Mosaic GPU] Add a result to the WGMMA op definition in the MLIR dialect
CI - Bazel CUDA tests (RBE) #82: Pull request #26029 synchronize by copybara-service bot
January 23, 2025 11:10 11m 1s test_718300974
January 23, 2025 11:10 11m 1s
#sdy unskip JAX Shardy tests that are already passing
CI - Bazel CUDA tests (RBE) #81: Pull request #26055 opened by copybara-service bot
January 23, 2025 11:05 7m 18s test_718786627
January 23, 2025 11:05 7m 18s
[Mosaic GPU] Add a result to the WGMMA op definition in the MLIR dialect
CI - Bazel CUDA tests (RBE) #80: Pull request #26029 synchronize by copybara-service bot
January 23, 2025 10:56 10m 6s test_718300974
January 23, 2025 10:56 10m 6s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI - Bazel CUDA tests (RBE) #79: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 10:48 6m 37s test_717844490
January 23, 2025 10:48 6m 37s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI - Bazel CUDA tests (RBE) #78: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 10:31 8m 16s test_717844490
January 23, 2025 10:31 8m 16s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI - Bazel CUDA tests (RBE) #77: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 09:49 6m 38s test_717844490
January 23, 2025 09:49 6m 38s
ProTip! You can narrow down the results and go further in time using created:<2025-01-23 or the other filters available.