Skip to content

Actions: pytorch/xla

Actions

Build upstream image

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
build_upstream_image.yml
698 workflow runs
698 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

Refactor Computation of jax into creation and call (#8789) Build upstream image #673: Commit 53b3ab8 pushed by tengyifei
March 12, 2025 03:18 7m 16s master
master
March 12, 2025 03:18 7m 16s
Remove pages_per_seq padding constrains and add MHA support (#8816) Build upstream image #672: Commit af16c21 pushed by yaochengji
March 12, 2025 02:57 10m 24s master
master
March 12, 2025 02:57 10m 24s
Fix the ragged kernel bug (#8806) Build upstream image #671: Commit 6804c65 pushed by vanbasten23
March 10, 2025 16:46 10m 28s master
master
March 10, 2025 16:46 10m 28s
Document xla override (#8802) Build upstream image #670: Commit 0407d6d pushed by tengyifei
March 9, 2025 21:59 10m 18s master
master
March 9, 2025 21:59 10m 18s
Leverage placeholder tensors for GA API (#8808) Build upstream image #669: Commit 5b7462c pushed by tengyifei
March 9, 2025 21:53 10m 29s master
master
March 9, 2025 21:53 10m 29s
Manually register einsum on xla (#8801) Build upstream image #668: Commit 1db6c86 pushed by tengyifei
March 7, 2025 03:56 10m 31s master
master
March 7, 2025 03:56 10m 31s
Re-enable triton tests. (#8803) Build upstream image #667: Commit 2c70a1c pushed by ysiraichi
March 6, 2025 20:13 10m 21s master
master
March 6, 2025 20:13 10m 21s
Lower _conj_copy and alias operation. (#8686) Build upstream image #666: Commit 6c53a1e pushed by tengyifei
March 6, 2025 02:44 10m 13s master
master
March 6, 2025 02:44 10m 13s
Fix ragged paged attention v2 to resolve the recompilation issue (#8797) Build upstream image #665: Commit d06a9c9 pushed by yaochengji
March 6, 2025 00:22 11m 17s master
master
March 6, 2025 00:22 11m 17s
Pin update to 20250303 (#8788) Build upstream image #664: Commit 9b61c1a pushed by tengyifei
March 5, 2025 21:04 10m 34s master
master
March 5, 2025 21:04 10m 34s
Use placeholder tensor in scan (#8785) Build upstream image #663: Commit a927928 pushed by tengyifei
March 5, 2025 20:24 10m 24s master
master
March 5, 2025 20:24 10m 24s
Use .backward() with in-place grad mutations for the GA API (#8768) Build upstream image #662: Commit 4540d81 pushed by bhavya01
March 5, 2025 20:01 10m 27s master
master
March 5, 2025 20:01 10m 27s
Integrate ragged paged attention v2 (#8791) Build upstream image #661: Commit 5644f44 pushed by yaochengji
March 5, 2025 18:33 10m 24s master
master
March 5, 2025 18:33 10m 24s
Minimal support for calling JAX from PyTorch/XLA (#8781) Build upstream image #660: Commit 17270e2 pushed by tengyifei
March 4, 2025 05:15 10m 22s master
master
March 4, 2025 05:15 10m 22s
Build torch_xla wheel in build script (#8782) Build upstream image #659: Commit 00fac78 pushed by tengyifei
March 4, 2025 03:36 11m 0s master
master
March 4, 2025 03:36 11m 0s
write _shard_map; refactor flash attention to support 5d inputs. (#8730) Build upstream image #658: Commit cee0820 pushed by qihqi
March 3, 2025 17:14 10m 26s master
master
March 3, 2025 17:14 10m 26s
Introduce a GRU module implemented with scan (#8777) Build upstream image #657: Commit 0017ff1 pushed by qihqi
March 3, 2025 05:42 10m 28s master
master
March 3, 2025 05:42 10m 28s
Fix build on Python 3.9 (#8759) Build upstream image #656: Commit c76e949 pushed by tengyifei
February 28, 2025 23:15 10m 18s master
master
February 28, 2025 23:15 10m 18s
Misc changes to make torchax runnable on GPU. (#8756) Build upstream image #655: Commit 60001c7 pushed by qihqi
February 28, 2025 22:01 10m 23s master
master
February 28, 2025 22:01 10m 23s
[scan] Make sure inputs into fn are not device_data IR nodes (#8769) Build upstream image #654: Commit 2feb0ac pushed by tengyifei
February 28, 2025 21:41 10m 26s master
master
February 28, 2025 21:41 10m 26s
Add sm_scale in ragged attention kernel (#8771) Build upstream image #653: Commit 2675e68 pushed by yaochengji
February 28, 2025 07:21 10m 36s master
master
February 28, 2025 07:21 10m 36s
Revert "Change num_seqs type from int to torch.Tensor" (#8767) Build upstream image #652: Commit 2e4f073 pushed by yaochengji
February 27, 2025 23:12 10m 4s master
master
February 27, 2025 23:12 10m 4s
Enable default buffer donation for gradient accumulation (#8758) Build upstream image #651: Commit 6712eb9 pushed by bhavya01
February 27, 2025 22:39 10m 26s master
master
February 27, 2025 22:39 10m 26s
Change num_seqs type from int to torch.Tensor (#8736) Build upstream image #650: Commit b18a65f pushed by yaochengji
February 27, 2025 18:39 10m 31s master
master
February 27, 2025 18:39 10m 31s
Lower as_strided_copy use fast path with slice (#8734) Build upstream image #649: Commit 1ab8216 pushed by tengyifei
February 27, 2025 02:45 12m 7s master
master
February 27, 2025 02:45 12m 7s