Skip to content

Actions: jax-ml/jax-triton

Actions

ci

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
ci.yml
91 workflow runs
91 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Triton integrate 2023-12-18 ci #139: Commit 4a5791d pushed by copybara-service bot
January 5, 2024 19:09 32s main
main
January 5, 2024 19:09 32s
Import new version of Triton ci #138: Commit 7778c47 pushed by copybara-service bot
December 15, 2023 10:57 36s main
main
December 15, 2023 10:57 36s
Import new version of Triton ci #137: Commit f0beb4c pushed by copybara-service bot
December 13, 2023 09:24 34s main
main
December 13, 2023 09:24 34s
[jax_triton] Allow user to override the custom call target name. ci #136: Commit 68c6262 pushed by copybara-service bot
November 7, 2023 19:25 53s main
main
November 7, 2023 19:25 53s
Expose more Triton compiler tuning options. ci #135: Commit b81596b pushed by copybara-service bot
November 1, 2023 00:28 47s main
main
November 1, 2023 00:28 47s
jax_triton: migrate from deprecated jax.linear_util to jax.extend.lin… ci #134: Commit 1f41ec2 pushed by copybara-service bot
October 26, 2023 21:54 50s main
main
October 26, 2023 21:54 50s
Bumping Triton version ci #133: Commit 15a479e pushed by copybara-service bot
October 19, 2023 20:36 33s main
main
October 19, 2023 20:36 33s
Updated jax.config import ci #132: Commit 25f4b55 pushed by copybara-service bot
October 19, 2023 18:07 39s main
main
October 19, 2023 18:07 39s
Testing triton integration 2023-09-14 ci #131: Commit f41e408 pushed by copybara-service bot
October 12, 2023 07:36 38s main
main
October 12, 2023 07:36 38s
When cuda is not available, the attributes `triton_kernel_call_lib.Tr… ci #130: Commit 06d3517 pushed by copybara-service bot
September 25, 2023 21:49 36s main
main
September 25, 2023 21:49 36s
Align the custom_call implementation in mlir and hlo_helpers. ci #129: Commit 170511d pushed by copybara-service bot
September 3, 2023 20:07 46s main
main
September 3, 2023 20:07 46s
Fix type annotation for triton_call args. ci #128: Commit c70d4ed pushed by copybara-service bot
August 31, 2023 21:35 43s main
main
August 31, 2023 21:35 43s
[jax_triton] Always use "triton_kernel_call" as custom call target name. ci #127: Commit 321a19e pushed by copybara-service bot
August 22, 2023 15:48 40s main
main
August 22, 2023 15:48 40s
[jax_triton] Add user-specified name field to serialized format. ci #126: Commit ad44ac8 pushed by copybara-service bot
August 16, 2023 09:53 35s main
main
August 16, 2023 09:53 35s
[JAX] Replace uses of jnp.array in types with jnp.ndarray. ci #125: Commit a25b1ba pushed by copybara-service bot
August 9, 2023 20:46 50s main
main
August 9, 2023 20:46 50s
Merge pull request #216 from jax-ml:sharadmv-patch-1 ci #124: Commit 75d47fb pushed by copybara-service bot
August 3, 2023 20:24 38s main
main
August 3, 2023 20:24 38s
[Pallas] Upstream pallas to JAX ci #123: Commit d258f8f pushed by copybara-service bot
August 1, 2023 23:43 36s main
main
August 1, 2023 23:43 36s
[jax_triton] Add support for float scalar inputs. ci #122: Commit b1b11d0 pushed by copybara-service bot
July 29, 2023 05:57 37s main
main
July 29, 2023 05:57 37s
Add Mosaic lowering for Pallas ci #121: Commit c6fc7b0 pushed by copybara-service bot
July 28, 2023 23:01 43s main
main
July 28, 2023 23:01 43s
* Test is flaky ci #120: Commit 46991ed pushed by copybara-service bot
July 27, 2023 20:39 48s main
main
July 27, 2023 20:39 48s
Update the code to not use the deprecated Chex assertions. ci #119: Commit bbf4287 pushed by copybara-service bot
July 19, 2023 16:57 39s main
main
July 19, 2023 16:57 39s
[pallas] Add support for custom call target name and serialized metad… ci #118: Commit 588ad6c pushed by copybara-service bot
July 19, 2023 16:54 38s main
main
July 19, 2023 16:54 38s
Separate out and add tests for Pallas indexing logic. ci #117: Commit 695afcf pushed by copybara-service bot
July 17, 2023 23:12 45s main
main
July 17, 2023 23:12 45s
Separate out and add tests for Pallas indexing logic. ci #116: Commit 163e1d4 pushed by copybara-service bot
July 17, 2023 19:28 40s main
main
July 17, 2023 19:28 40s
Fix vmap rules when num_index_operands > 0 ci #115: Commit 1966b57 pushed by copybara-service bot
July 17, 2023 17:25 50s main
main
July 17, 2023 17:25 50s