pytorch/xla

Pull Requests Commits

fix libtpu path (#9008)

zpcore committed 281 days ago

Verified e9cb086b

Remove config warning log from GradAcc API (#9006)

rpsilva-aws committed 281 days ago

Verified b20641e9

Guide to debugging PyTorch C++ (#8998)

yaoshiang committed 282 days ago

Verified 39aef3a1

Replace upstream GRU implementation with scan-based GRU (#9010)

iwknow committed 283 days ago

Verified faf777a2

update to r2.7 rc5 (#9001)

zpcore committed 284 days ago

Verified 329be9e5

Default explicit donation for step barriers (#8982)

rpsilva-aws committed 284 days ago

Verified e2f2852e

typo fix (#8990)

haifeng-jin committed 284 days ago

Verified 428c86d2

scan-based GRU falls back to nn.GRU when `bidirectional` is true. (#8984)

iwknow committed 284 days ago

Verified 0239d8c5

Fix the source param sharding for GradAcc API (#8999)

rpsilva-aws committed 284 days ago

Verified db49eddb

Set scoped vmem for paged attention (#8988)

zpcore committed 285 days ago

Verified 366f248a

Fix merge conflict (#8991)

zpcore committed 285 days ago

Verified 083934fd

Add an option for JittableModule to dedup parameters. (#8965)

qihqi committed 286 days ago

Verified cce7cc07

Add a helper class to handle mesh and sharding (#8967)

qihqi committed 286 days ago

Verified 78824755

Fix splash attention test (#8978)

zpcore committed 286 days ago

Verified 13155c9c

Fix call_jax hashing (#8981)

zpcore committed 286 days ago

Verified a48d6a97

Update to ubuntu latest (#8979)

qihqi committed 286 days ago

Verified a37a5852

@assume_pure (#8962)

tengyifei committed 286 days ago

Verified cf764ffa

Avoid re-computing computation hashes (#8976)

rpsilva-aws committed 287 days ago

Verified bcca5eba

Disable one splash attention test (#8970)

zpcore committed 287 days ago

Verified ba5cd4cb

Add tuned parameters for Qwen/Qwen2.5-32B (#8966)

yarongmu-google committed 290 days ago

Verified 4eea5e13

Adapt Splash Attention from TorchPrime (#8911)

zpcore committed 291 days ago

Verified 45830519

Make scan-based GRU support `batch_first` parameter. (#8964)

iwknow committed 291 days ago

Verified aad87e61

[call_jax] support returning PyTree from the JAX function (#8957)

tengyifei committed 291 days ago

Verified 297f5de6

Remove deprecated typing module in ansible flow (#8955)

tengyifei committed 292 days ago

Verified e77bd2f6

Add test to ensure scan-based and the standard GRU are interchangeable. (#8949)

iwknow committed 293 days ago

Verified d4d997dc

Update to pytorch 2.6 (#8944)

qihqi committed 293 days ago

Verified 085f8bc1

Pin update to 20250406 (#8945)

tengyifei committed 294 days ago

Verified 0837fac3

Use pages_per_seq * page_size instead of directly passing max_model_len (#8950)

yaochengji committed 294 days ago

Verified 9ec626e6

Fix ragged_paged_attention op signature (#8943)

yaochengji committed 295 days ago

Verified 80929a91

add rc4 trigger (#8941)

zpcore committed 296 days ago

Verified ac9a39f4

Newer Older