xla
Follow up on ragged kernel wrapper
#8737
Open

Follow up on ragged kernel wrapper #8737

vanbasten23 wants to merge 4 commits into master from xiowei/follow_up_on_ragged_kernel_wrapper
vanbasten23
vanbasten23 create the test
51c48ec3
vanbasten23 avoid cpu and tpu back and forth
3c4eda73
vanbasten23 add a few more tests. Somehow the dynamo test fails
adcaa1be
vanbasten23 add more tests
44d00ab3

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone