Skip to content

Actions: jax-ml/jax

CI

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
27,997 workflow runs
27,997 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[Mosaic GPU] Remove an unnecessary restriction in the vector.store
CI #70754: Commit 6f60992 pushed by copybara-service bot
January 23, 2025 12:24 11m 49s main
January 23, 2025 12:24 11m 49s
[Mosaic GPU] Remove an unnecessary restriction in the vector.store lowering
CI #70753: Pull request #26035 synchronize by copybara-service bot
January 23, 2025 12:08 10m 15s test_718385442
January 23, 2025 12:08 10m 15s
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
CI #70752: Pull request #26032 synchronize by copybara-service bot
January 23, 2025 11:38 10m 4s test_718302204
January 23, 2025 11:38 10m 4s
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
CI #70751: Commit f57d603 pushed by copybara-service bot
January 23, 2025 11:38 10m 32s main
January 23, 2025 11:38 10m 32s
[Mosaic GPU] Simplify enums in the MLIR Mosaic GPU dialect.
CI #70750: Pull request #26032 synchronize by copybara-service bot
January 23, 2025 11:24 10m 21s test_718302204
January 23, 2025 11:24 10m 21s
[Mosaic GPU] Add a result to the WGMMA op definition in the MLIR dialect
CI #70749: Pull request #26029 synchronize by copybara-service bot
January 23, 2025 11:10 10m 47s test_718300974
January 23, 2025 11:10 10m 47s
January 23, 2025 11:10 10m 25s
#sdy unskip JAX Shardy tests that are already passing
CI #70747: Pull request #26055 opened by copybara-service bot
January 23, 2025 11:05 10m 14s test_718786627
January 23, 2025 11:05 10m 14s
[Mosaic GPU] Add a result to the WGMMA op definition in the MLIR dialect
CI #70746: Pull request #26029 synchronize by copybara-service bot
January 23, 2025 10:56 10m 4s test_718300974
January 23, 2025 10:56 10m 4s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI #70745: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 10:48 10m 16s test_717844490
January 23, 2025 10:48 10m 16s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI #70744: Commit 3a411d8 pushed by copybara-service bot
January 23, 2025 10:48 10m 21s main
January 23, 2025 10:48 10m 21s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI #70743: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 10:31 10m 18s test_717844490
January 23, 2025 10:31 10m 18s
[Mosaic GPU] Implement basic WGMMAFragLayout inference and propagation
CI #70742: Pull request #26007 synchronize by copybara-service bot
January 23, 2025 09:49 6m 49s test_717844490
January 23, 2025 09:49 6m 49s
[Mosaic GPU] Add basic support for TMA with sub-byte types
CI #70740: Pull request #26012 synchronize by copybara-service bot
January 23, 2025 08:16 10m 8s test_717952238
January 23, 2025 08:16 10m 8s
Fix typo and improve readability of workflow documentation
CI #70739: Pull request #26054 opened by copybara-service bot
January 23, 2025 04:57 10m 26s test_718687264
January 23, 2025 04:57 10m 26s
[JAX] Optimize array shard reordering
CI #70738: Pull request #26042 synchronize by copybara-service bot
January 23, 2025 03:29 10m 28s test_718482553
January 23, 2025 03:29 10m 28s
January 23, 2025 03:14 10m 14s
January 23, 2025 01:51 11m 37s
Do not check device_count constraint for some users.
CI #70732: Pull request #26025 synchronize by copybara-service bot
January 23, 2025 00:59 10m 29s test_716346724
January 23, 2025 00:59 10m 29s
[sharding_in_types] Make vmap work with shard_map + pallas
CI #70731: Commit 704b2e5 pushed by copybara-service bot
January 23, 2025 00:48 9m 50s main
January 23, 2025 00:48 9m 50s
[sharding_in_types] Make vmap work with shard_map + pallas
CI #70730: Pull request #26047 synchronize by copybara-service bot
January 23, 2025 00:47 10m 14s test_718511907
January 23, 2025 00:47 10m 14s