pytorch/xla

Pull Requests Commits

Migration from deprecated (@openxla/head) headers and build-targets (#9033)

sdasgup3 committed 302 days ago

Verified 047de24d

Add runtime check when using non-kernel for ragged paged attn (#8958)

bythew3i committed 302 days ago

Verified be257b54

cpp_debugger.md updates (#9034)

mikegre-google committed 302 days ago

Verified fc43df2d

Propagate mutations of Tensor slices to the source Tensor. (#9025)

wenxindongwork committed 302 days ago

Verified 844b5b74

Update README for 2.7 release (#9031)

zpcore committed 303 days ago

Verified 8d42e9a2

Update CONTRIBUTING.md (#9030)

haifeng-jin committed 303 days ago

Verified 1b6630e8

Fix broken link to GPU docs (#9023)

ghpvnist committed 303 days ago

Verified 50ec9a66

Trigger final wheel (#9029)

zpcore committed 304 days ago

Verified eff4d644

Fix link to show rendered markdown instead of edit (#9028)

haifeng-jin committed 304 days ago

Verified 615f8a17

[cleanup] Remove install_post_deps_pytorch_xla (#9027)

tengyifei committed 304 days ago

Verified 419f7250

disable tests using mark_sharding + assume_pure on GPU (#9013)

qihqi committed 304 days ago

Verified a8653c93

Remove trailing spaces from CONTRIBUTING.md (#9024)

ghpvnist committed 305 days ago

Verified 1b7422d0

Typo fix for OP_LOWERING_GUIDE (#9020)

haifeng-jin committed 305 days ago

Verified 50d352f6

handle requires_grad in torchax (#8992)

qihqi committed 306 days ago

Verified f177e762

composibility of assume_pure and call_jax (#8989)

qihqi committed 306 days ago

Verified 8e6a5e5d

fix libtpu path (#9008)

zpcore committed 307 days ago

Verified e9cb086b

Remove config warning log from GradAcc API (#9006)

rpsilva-aws committed 307 days ago

Verified b20641e9

Guide to debugging PyTorch C++ (#8998)

yaoshiang committed 307 days ago

Verified 39aef3a1

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

iwknow committed 308 days ago

Verified faf777a2

update to r2.7 rc5 (#9001)

zpcore committed 310 days ago

Verified 329be9e5

Default explicit donation for step barriers (#8982)

rpsilva-aws committed 310 days ago

Verified e2f2852e

typo fix (#8990)

haifeng-jin committed 310 days ago

Verified 428c86d2

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

iwknow committed 310 days ago

Verified 0239d8c5

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

rpsilva-aws committed 310 days ago

Verified db49eddb

Set scoped vmem for paged attention (#8988)

zpcore committed 311 days ago

Verified 366f248a

Fix merge conflict (#8991)

zpcore committed 311 days ago

Verified 083934fd

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

qihqi committed 311 days ago

Verified cce7cc07

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

qihqi committed 311 days ago

Verified 78824755

Fix splash attention test (#8978)

zpcore committed 311 days ago

Verified 13155c9c

Fix call_jax hashing (#8981)

zpcore committed 311 days ago

Verified a48d6a97

Newer Older