xla
Fallback `_embedding_bag_backward` and force `sparse=false`.
#7584
Merged
Overview
Commits
11
Changes
View On GitHub

Fallback `_embedding_bag_backward` and force `sparse=false`. #7584

ysiraichi merged 11 commits into master from ysiraichi/fallback-embedding-bag-backward
ysiraichi
ysiraichi ysiraichi added xla:gpu
ysiraichi ysiraichi requested a review from bhavya01 bhavya01 1 year ago
ysiraichi ysiraichi requested a review from JackCaoG JackCaoG 1 year ago
JackCaoG
JackCaoG commented on 2024-06-27
ysiraichi ysiraichi force pushed from cf3a4a39 to 6e5131af 1 year ago
ysiraichi Fallback on `_embedding_bag_backward`.
516228c7
ysiraichi Add torch_pin.
5ffaadd9
ysiraichi ysiraichi force pushed from 6e5131af to 5ffaadd9 1 year ago
ysiraichi Add test.
dc6ab01b
bhavya01
bhavya01 approved these changes on 2024-06-28
ysiraichi Warn only if sparse.
b59e5b0d
vanbasten23
vanbasten23 commented on 2024-07-01
vanbasten23
vanbasten23 approved these changes on 2024-07-01
ysiraichi Remove torch_pin.
46371e53
ysiraichi Propagate fallback function name change.
a35b53a1
ysiraichi Fix lint issues.
6c3f7788
ysiraichi Add path for `XLA_DISABLE_FUNCTIONALIZATION=1`.
9f052000
ysiraichi Fix fallback function rename.
e85be64a
ysiraichi Fix lint issues.
c87e76a9
ysiraichi Fix lint issues.
8d61cc11
ysiraichi ysiraichi merged 3bcb1fb1 into master 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
xla:gpu
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum