Skip to content

Actions: pytorch/xla

Actions

torchax

Actions

Loading...
Loading

Show workflow options

Create status badge

Loading
torchax.yml
1,275 workflow runs
1,275 workflow runs

Filter by Event

Filter by Status

Filter by Branch

Filter by Actor

update l_ref in kernel matrix calculation torchax #1203: Pull request #9372 synchronize by Chenyaaang
June 16, 2025 21:45 15m 3s Chenyaaang:vmem-oom-fix
Chenyaaang:vmem-oom-fix
June 16, 2025 21:45 15m 3s
update l_ref in kernel matrix calculation torchax #1202: Pull request #9372 opened by Chenyaaang
June 16, 2025 21:23 14m 58s Chenyaaang:vmem-oom-fix
Chenyaaang:vmem-oom-fix
June 16, 2025 21:23 14m 58s
Clarify xr.device_type() API and use them torchax #1201: Pull request #9343 synchronize by ghpvnist
June 16, 2025 20:41 15m 44s ghpvnist:device
ghpvnist:device
June 16, 2025 20:41 15m 44s
Lower clamp torchax #1200: Pull request #9123 synchronize by ghpvnist
June 16, 2025 20:41 15m 30s ghpvnist:clamp
ghpvnist:clamp
June 16, 2025 20:41 15m 30s
Migrate torch_xla.device() to torch.device('xla') torchax #1199: Pull request #9253 synchronize by ghpvnist
June 16, 2025 20:40 15m 15s ghpvnist:9252
ghpvnist:9252
June 16, 2025 20:40 15m 15s
Avoid recompilation caused by scan_layers (#9367) torchax #1198: Commit 41b9e45 pushed by tengyifei
June 16, 2025 20:12 16m 8s master
master
June 16, 2025 20:12 16m 8s
manual revert of 9346 torchax #1197: Pull request #9369 opened by bfolie
June 16, 2025 19:16 15m 23s bfolie/revert-9368
bfolie/revert-9368
June 16, 2025 19:16 15m 23s
Revert "Update ragged paged attention kernel to prevent vmem oom" torchax #1196: Pull request #9368 opened by bfolie
June 16, 2025 18:45 15m 0s revert-9346-vmem-oom
revert-9346-vmem-oom
June 16, 2025 18:45 15m 0s
Avoid recompilation caused by scan_layers torchax #1195: Pull request #9367 opened by tengyifei
June 16, 2025 18:42 15m 21s yifeit/no-dangling-tensor
yifeit/no-dangling-tensor
June 16, 2025 18:42 15m 21s
fix all_gather_into_tensor test and logic (#9332) torchax #1194: Commit 9b0b02f pushed by bfolie
June 16, 2025 18:29 15m 31s master
master
June 16, 2025 18:29 15m 31s
Support torch.distributed.scatter collective torchax #1193: Pull request #9365 synchronize by bfolie
June 16, 2025 17:10 15m 58s bfolie/9314-scatter
bfolie/9314-scatter
June 16, 2025 17:10 15m 58s
Improve assume_pure SPMD functionality (#9360) torchax #1192: Commit 0be0fbd pushed by tengyifei
June 16, 2025 17:07 15m 30s master
master
June 16, 2025 17:07 15m 30s
Support torch.distributed.scatter collective torchax #1191: Pull request #9365 opened by bfolie
June 16, 2025 17:01 9m 7s bfolie/9314-scatter
bfolie/9314-scatter
June 16, 2025 17:01 9m 7s
fix all_gather_into_tensor test and logic torchax #1190: Pull request #9332 synchronize by bfolie
June 16, 2025 16:16 15m 11s bfolie/all-gather-into-tensor
bfolie/all-gather-into-tensor
June 16, 2025 16:16 15m 11s
Add autocast feature as torchax.amp.autocast. torchax #1189: Pull request #9364 synchronize by qihqi
June 16, 2025 15:44 15m 59s hanq_autocast_stopgap
hanq_autocast_stopgap
June 16, 2025 15:44 15m 59s
Add autocast feature as torchax.amp.autocast. torchax #1188: Pull request #9364 opened by qihqi
June 16, 2025 15:39 5m 33s hanq_autocast_stopgap
hanq_autocast_stopgap
June 16, 2025 15:39 5m 33s
Improve assume_pure SPMD functionality torchax #1187: Pull request #9360 synchronize by tengyifei
June 16, 2025 06:03 15m 27s yifeit/nested-pspec
yifeit/nested-pspec
June 16, 2025 06:03 15m 27s
[WIP] Add autocast in torchax torchax #1186: Pull request #9361 synchronize by qihqi
June 16, 2025 04:22 3m 5s hanq_autocast
hanq_autocast
June 16, 2025 04:22 3m 5s
[WIP] Add autocast in torchax torchax #1185: Pull request #9361 synchronize by qihqi
June 16, 2025 03:36 1m 57s hanq_autocast
hanq_autocast
June 16, 2025 03:36 1m 57s
[WIP] Add autocast in torchax torchax #1184: Pull request #9361 synchronize by qihqi
June 16, 2025 02:12 2m 28s hanq_autocast
hanq_autocast
June 16, 2025 02:12 2m 28s
[WIP] Add autocast in torchax torchax #1183: Pull request #9361 synchronize by qihqi
June 16, 2025 02:10 1m 36s hanq_autocast
hanq_autocast
June 16, 2025 02:10 1m 36s
Add an option to not use dlpack. (#9304) torchax #1182: Commit 55a7540 pushed by qihqi
June 16, 2025 02:08 14m 50s master
master
June 16, 2025 02:08 14m 50s
[WIP] Add autocast in torchax torchax #1181: Pull request #9361 opened by qihqi
June 16, 2025 02:07 2m 23s hanq_autocast
hanq_autocast
June 16, 2025 02:07 2m 23s
Improve assume_pure SPMD functionality torchax #1180: Pull request #9360 synchronize by tengyifei
June 16, 2025 01:21 15m 37s yifeit/nested-pspec
yifeit/nested-pspec
June 16, 2025 01:21 15m 37s
Improve assume_pure SPMD functionality torchax #1179: Pull request #9360 synchronize by tengyifei
June 16, 2025 01:17 4m 3s yifeit/nested-pspec
yifeit/nested-pspec
June 16, 2025 01:17 4m 3s