Skip to content

Actions: jax-ml/jax

Actions

JAX Array API

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
jax-array-api.yml
11,177 workflow runs
11,177 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

[pallas:mosaic_gpu] Use emit_pipeline for pipelining in the lowering JAX Array API #11658: Pull request #26561 synchronize by copybara-service bot
February 25, 2025 10:54 2m 55s test_726053925
test_726053925
February 25, 2025 10:54 2m 55s
[pallas:mosaic_gpu] Use emit_pipeline for pipelining in the lowering JAX Array API #11657: Pull request #26561 synchronize by copybara-service bot
February 25, 2025 10:47 2m 57s test_726053925
test_726053925
February 25, 2025 10:47 2m 57s
[Pallas:MGPU] Enable lowering for .astype and scalar broadcasts JAX Array API #11656: Pull request #26731 synchronize by copybara-service bot
February 25, 2025 10:46 2m 58s test_730794575
test_730794575
February 25, 2025 10:46 2m 58s
[Pallas:MGPU] Add support for WG semantics lowering of scan_p JAX Array API #11655: Pull request #26732 opened by copybara-service bot
February 25, 2025 10:42 2m 47s test_730794577
test_730794577
February 25, 2025 10:42 2m 47s
[Pallas:MGPU] Enable lowering for .astype and scalar broadcasts JAX Array API #11654: Pull request #26731 opened by copybara-service bot
February 25, 2025 10:41 2m 56s test_730794575
test_730794575
February 25, 2025 10:41 2m 56s
Upgrade to Bazel 7.4.1 JAX Array API #11653: Pull request #26513 synchronize by copybara-service bot
February 25, 2025 10:29 3m 0s test_726036888
test_726036888
February 25, 2025 10:29 3m 0s
Skip tests using StringDType when NumPy version is below 2.0 JAX Array API #11652: Commit 7fb6788 pushed by copybara-service bot
February 25, 2025 10:25 3m 15s main
main
February 25, 2025 10:25 3m 15s
[Pallas:MGPU] Change WG semantics convention to represent scalar arra… JAX Array API #11651: Commit 71c7622 pushed by copybara-service bot
February 25, 2025 10:24 2m 1s main
main
February 25, 2025 10:24 2m 1s
[Mosaic GPU][NFC] Remove unnecessary lambda wrappers from test. JAX Array API #11650: Pull request #26727 synchronize by copybara-service bot
February 25, 2025 10:23 2m 53s test_730774851
test_730774851
February 25, 2025 10:23 2m 53s
[Pallas:MGPU] Avoid SMEM->GMEM wait if no outputs are transferred in … JAX Array API #11649: Commit d0d5bba pushed by copybara-service bot
February 25, 2025 10:22 1m 59s main
main
February 25, 2025 10:22 1m 59s
Remove MemoryEffects annotations from async_{load/store} ops JAX Array API #11648: Pull request #26723 synchronize by copybara-service bot
February 25, 2025 10:22 2m 58s test_730766772
test_730766772
February 25, 2025 10:22 2m 58s
[Mosaic GPU] Add dialect lowering logic for splat constants. JAX Array API #11647: Pull request #26726 synchronize by copybara-service bot
February 25, 2025 10:22 2m 53s test_730773931
test_730773931
February 25, 2025 10:22 2m 53s
[Pallas:MGPU] Avoid SMEM->GMEM wait if no outputs are transferred in the pipeline loop JAX Array API #11646: Pull request #26725 synchronize by copybara-service bot
February 25, 2025 10:21 2m 57s test_730769595
test_730769595
February 25, 2025 10:21 2m 57s
[Mosaic GPU] Add layout inference for arith.Ext{F,SI,UI}Op and arith.Trunc{F,I}Op. JAX Array API #11645: Pull request #26728 synchronize by copybara-service bot
February 25, 2025 10:21 2m 58s test_730777935
test_730777935
February 25, 2025 10:21 2m 58s
[Pallas:MGPU] Change WG semantics convention to represent scalar arrays using scalars JAX Array API #11644: Pull request #26730 synchronize by copybara-service bot
February 25, 2025 10:20 2m 51s test_730778637
test_730778637
February 25, 2025 10:20 2m 51s
[better_errors] Port the Pallas debug info mechanisms to the new JAX debugging info JAX Array API #11643: Pull request #26058 synchronize by gnecula
February 25, 2025 10:16 3m 11s gnecula:debug_info_pallas
gnecula:debug_info_pallas
February 25, 2025 10:16 3m 11s
[better_errors] Port the Pallas debug info mechanisms to the new JAX debugging info JAX Array API #11642: Pull request #26058 synchronize by gnecula
February 25, 2025 10:08 3m 2s gnecula:debug_info_pallas
gnecula:debug_info_pallas
February 25, 2025 10:08 3m 2s
Skip tests using StringDType when NumPy version is below 2.0 JAX Array API #11641: Pull request #26729 synchronize by copybara-service bot
February 25, 2025 10:03 2m 55s test_730777217
test_730777217
February 25, 2025 10:03 2m 55s
Update XLA dependency to use revision JAX Array API #11640: Commit 43f2da8 pushed by copybara-service bot
February 25, 2025 10:01 3m 2s main
main
February 25, 2025 10:01 3m 2s
Upgrade to Bazel 7.4.1 JAX Array API #11639: Pull request #26513 synchronize by copybara-service bot
February 25, 2025 10:00 3m 8s test_726036888
test_726036888
February 25, 2025 10:00 3m 8s
Skip tests using StringDType when NumPy version is below 2.0 JAX Array API #11638: Pull request #26729 synchronize by copybara-service bot
February 25, 2025 09:34 2m 55s test_730777217
test_730777217
February 25, 2025 09:34 2m 55s
[pallas:mosaic_gpu] Added WG lowering rules for TMA primitives and ru… JAX Array API #11637: Commit 7eadc64 pushed by copybara-service bot
February 25, 2025 09:32 3m 3s main
main
February 25, 2025 09:32 3m 3s
[pallas:mosaic_gpu] Added WG lowering rules for TMA primitives and run_scoped_p JAX Array API #11636: Pull request #26714 synchronize by copybara-service bot
February 25, 2025 09:31 2m 57s test_729896457
test_729896457
February 25, 2025 09:31 2m 57s
[Pallas:MGPU] Change WG semantics convention to represent scalar arrays using scalars JAX Array API #11635: Pull request #26730 opened by copybara-service bot
February 25, 2025 09:26 2m 55s test_730778637
test_730778637
February 25, 2025 09:26 2m 55s
[Pallas:MGPU] Consistently use i32 as the grid index type in emit_pip… JAX Array API #11634: Commit 80848ad pushed by copybara-service bot
February 25, 2025 09:24 3m 3s main
main
February 25, 2025 09:24 3m 3s