xla
custom autograd op redispatching fix
#5600
Merged

custom autograd op redispatching fix #5600

JackCaoG merged 2 commits into master from dispatcher_custom_fix
bdhirsh
bdhirsh custom autograd op redispatching fix
6e69ee5b
bdhirsh bdhirsh requested a review from JackCaoG JackCaoG 2 years ago
bdhirsh only add TLSSnapshot when Python key is in the keyset
6c6306af
JackCaoG
JackCaoG commented on 2023-09-18
JackCaoG
JackCaoG approved these changes on 2023-09-18
JackCaoG JackCaoG merged 5dd9108a into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone