xla
Cache HLO in xb.call_jax and support non-tensor args
#8878

Merged

Cache HLO in xb.call_jax and support non-tensor args #8878

tengyifei merged 2 commits into master from yifeit/call-jax-cache

Cache HLO in xb.call_jax and support non-tensor args

d95c7a4d

tengyifei marked this pull request as ready for review 335 days ago

tengyifei requested a review from

qihqi 335 days ago

tengyifei requested a review from

zpcore 335 days ago

tengyifei requested a review from

bhavya01 335 days ago

qihqi commented on 2025-03-24

qihqi approved these changes on 2025-03-24

Address comments

1eb8f16f

tengyifei merged a3ef52e5 into master 335 days ago

zpcore commented on 2025-03-31

Reviewers

qihqi

zpcore

bhavya01

Assignees

No one assigned

Labels

None yet

Milestone

No milestone