Skip to content

Pull requests: pytorch/xla

New pull request New
156 Open 6,115 Closed
156 Open 6,115 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Follow up on ragged kernel wrapper
#8737 opened Feb 24, 2025 by vanbasten23 • Draft
Change num_seqs type from int to torch.Tensor
#8736 opened Feb 22, 2025 by vanbasten23 Loading…
7
Add --network=host to TPU docker build command
#8735 opened Feb 21, 2025 by bhavya01 Loading…
@bhavya01
2
Lower as_strided_copy use fast path with slice
#8734 opened Feb 21, 2025 by zpcore Loading…
Improve the ragged kernel benchmarking script.
#8733 opened Feb 21, 2025 by vanbasten23 Loading…
3
write _shard_map; refactor flash attention to use it.
#8730 opened Feb 21, 2025 by qihqi Loading…
Extend buffer donation aliasing APIs
#8721 opened Feb 19, 2025 by rpsilva-aws Loading…
48
[SPMD] Add API to disable global SPMD config
#8717 opened Feb 18, 2025 by lsy323 Loading…
7
[SPMD] Add API to create global tensor from local CPU/TPU shards
#8716 opened Feb 18, 2025 by lsy323 • Draft
Document how to debug the dispatcher
#8712 opened Feb 15, 2025 by tengyifei Loading…
1
Add instruction for exporting inlined constant
#8707 opened Feb 13, 2025 by qihqi Loading…
4
Add 5D support for flash_attention
#8693 opened Feb 9, 2025 by qihqi Loading…
5
q
#8690 opened Feb 8, 2025 by test1githu Loading…
Lower _conj_copy and alias operation. op lowering
#8686 opened Feb 6, 2025 by ysiraichi Loading…
1
@ysiraichi
31
enable tensor.copy_ with torch native cpu tensor as source
#8682 opened Feb 6, 2025 by yaochengji Loading…
2
Transition to Hermetic CUDA
#8665 opened Feb 3, 2025 by ysiraichi • Draft
@ysiraichi
2
add aarch64 platform build support
#8663 opened Feb 2, 2025 by snadampal Loading…
Port infra from tpu-pytorch to tpu-pytorch-releases and multi TPU support CI CI related change
#8574 opened Jan 15, 2025 by zpcore Loading…
@zpcore
Lower cummin op op lowering
#8565 opened Jan 14, 2025 by zyy-martin Loading…
1
@zyy-martin
12
Implement torch.linspace and torch.logspace
#8533 opened Jan 6, 2025 by vyom1611 Loading…
6
Lowering Aten op to composite op instead of small ops
#8502 opened Dec 18, 2024 by Zantares Loading…
20
Support S32 index operations for copy and fill
#8463 opened Dec 6, 2024 by rpsilva-aws Loading…
3
update test_replace_xla_tensor
#8458 opened Dec 5, 2024 by laithsakka Loading…
Remove the constraint of pages_per_sequence%num_kv_pages_per_compute_…
#8457 opened Dec 5, 2024 by vanbasten23 • Draft
[WIP] Update test_ops.py
#8449 opened Dec 4, 2024 by ManfeiBai • Draft
ProTip! What’s not been updated in a month: updated:<2025-01-24.