Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support S32/U32 indices for BWD embedding and index ops & Neuron implicit downcast #8450

Closed
wants to merge 17 commits into from
Closed

Support S32/U32 indices for BWD embedding and index ops & Neuron implicit downcast #8450

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
2c735b8
Extend operation indices to int32
rpsilva-aws Dec 4, 2024
de2ff40
Neuron S64/U64 downcast
rpsilva-aws Dec 4, 2024
0e10764
Merge branch 'pytorch:master' into rpsilva_downcast
rpsilva-aws Dec 5, 2024
f3dadf3
Use regular `torch.Tensor` for CPU tensors (#8416)
qihqi Nov 27, 2024
c00bf53
Support SegmentID when doing data prallel SPMD (#8425)
JackCaoG Nov 28, 2024
dde47a7
Reenable the distributed checkpointing test (#8424)
JackCaoG Dec 2, 2024
eb40b02
Update `train_resnet_benchmark.py` reference (#8428)
emmanuel-ferdman Dec 3, 2024
af0773a
Switch to libtpu package from the libtpu-wheels registry (#8409)
tengyifei Dec 4, 2024
ddcb2b9
[scan] Add a test under SPMD (#8419)
tengyifei Dec 4, 2024
b14d01f
Align rrelu_with_noise schema to reflect noise mutation [branch] (#8363)
IvanKobzarev Dec 4, 2024
317d382
[Documentation] Added a section pointing readers to the AI-Hypercompu…
mikegre-google Dec 4, 2024
61a70f5
Updates Contributing.md (#8343)
mikegre-google Dec 4, 2024
5d575a8
Update torch version to 2.5.1 (#8452)
qihqi Dec 4, 2024
582801f
Fix an edge case issue when storing output in the paged attention. (#…
vanbasten23 Dec 5, 2024
4d4d9f2
DLPack: fix test using PyTorch API. (#8348)
ysiraichi Dec 5, 2024
f00155f
Merge branch 'rpsilva_downcast' of github.com:rpsilva-aws/xla into rp…
rpsilva-aws Dec 6, 2024
8c25d80
Fix parameterized tests
rpsilva-aws Dec 6, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading