xla
Support S32/U32 indices for BWD embedding & Neuron implicit downcast
#8462

Merged

Support S32/U32 indices for BWD embedding & Neuron implicit downcast #8462

tengyifei merged 3 commits into pytorch:master from rpsilva-aws:rpsilva_downcast_v2

rpsilva-aws changed the title ~~Rpsilva downcast v2~~ Support S32/U32 indices for BWD embedding & Neuron implicit downcast 333 days ago

rpsilva-aws marked this pull request as ready for review 333 days ago

tengyifei added tpuci

tengyifei requested changes on 2024-12-06

rpsilva-aws force pushed from d4b65d79 to f2a73f17 333 days ago

rpsilva-aws force pushed from c2fb7efe to 95d0f0cb 333 days ago

tengyifei approved these changes on 2024-12-06

rpsilva-aws force pushed from 95d0f0cb to 75f37b02 333 days ago

Align S32 embedding indices with PyTorch

b7f16e25

Implicit S364/U32 downcasting for Neuron

14546088

Separate Neuron specific tests

a2109204

rpsilva-aws force pushed from 75f37b02 to a2109204 333 days ago

tengyifei approved these changes on 2024-12-06

tengyifei merged 00c0e966 into master 332 days ago

rpsilva-aws deleted the rpsilva_downcast_v2 branch 330 days ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone