xla
Introduce apply_xla_patch_to_nn_linear and test that in a scan
#8739

Merged

Introduce apply_xla_patch_to_nn_linear and test that in a scan #8739

tengyifei merged 1 commit into master from yifeit/workaround-einsum

tengyifei force pushed from c53acea7 to 7d834f7c 1 year ago

tengyifei changed the title ~~Support einsum layers in a scan~~ Introduce apply_xla_patch_to_nn_linear and test that in a scan 1 year ago

tengyifei marked this pull request as ready for review 1 year ago

Support einsum layers in a scan

2cf50c8b

tengyifei force pushed from 7d834f7c to 2cf50c8b 1 year ago

tengyifei requested a review from

bhavya01 1 year ago

tengyifei requested a review from

pgmoka 1 year ago

tengyifei requested a review from

zpcore 1 year ago

zpcore commented on 2025-02-25

zpcore approved these changes on 2025-02-25

tengyifei merged 6f020aad into master 1 year ago

pgmoka commented on 2025-02-25

Reviewers

zpcore

pgmoka

bhavya01

Assignees

No one assigned

Labels

None yet

Milestone

No milestone