xla
Helper function to wrap entry HLO
#3920
Merged
Overview
Commits
12
Changes
View On GitHub

Helper function to wrap entry HLO #3920

JackCaoG merged 12 commits into master from tuplify_parameters
JackCaoG
JackCaoG JackCaoG force pushed from cc2b4b43 to ceb994ec 3 years ago
JackCaoG
JackCaoG JackCaoG changed the title [DRAFT] Helper function to wrap entry HLO Helper function to wrap entry HLO 3 years ago
JackCaoG JackCaoG requested a review from will-cromar will-cromar 3 years ago
JackCaoG
JackCaoG
JackCaoG Helper function to wrap entry HLO
f5c7367b
JackCaoG Use the wrapped computation during computation
07161dce
JackCaoG don't tuple the wrapped hlo's result
51aecfbd
JackCaoG clean up
6890aa1f
JackCaoG preserve aliasing
a878fc6f
JackCaoG Add XLA_PARAMETER_WRAPPING_THREADSHOLD
53f9dc0b
JackCaoG code refactor
a80d4ffd
JackCaoG Add debug VLOG
a198ce5b
JackCaoG Add test
55797a72
JackCaoG clean up
2bd6bb22
JackCaoG clean up
bf9b61ec
JackCaoG JackCaoG force pushed from ca390fd7 to bf9b61ec 3 years ago
JackCaoG JackCaoG requested a review from alanwaketan alanwaketan 3 years ago
alanwaketan
alanwaketan approved these changes on 2022-08-26
JackCaoG address review comments
6c8763af
JackCaoG JackCaoG merged 41a4f855 into master 3 years ago
JackCaoG
ronghanghu
ronghanghu

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum