xla
Unify the return type of w8a8 matmul between fallback and the actual impl.
#9452

Merged

Unify the return type of w8a8 matmul between fallback and the actual impl. #9452

vanbasten23 merged 5 commits into master from xiowei/fix_nonxla_return_type

vanbasten23 requested a review from

yaochengji 323 days ago

vanbasten23 marked this pull request as ready for review 323 days ago

yaochengji approved these changes on 2025-07-08

vanbasten23 force pushed from e527723b to 9c2c0b2a 322 days ago

Unify the return type of w8a8 matmul across fallback and the actual i…

a1be787c

linter

b10868c4

make the function more readable

ea8cbd98

also check frobeneous error

4e46f5b4

Only run the test_quantized_matmul_int8_wrapper_fallback test on TPU

aac2352d

vanbasten23 force pushed from 8f363492 to aac2352d 322 days ago

vanbasten23 merged 52569ecc into master 321 days ago

Reviewers

yaochengji

Assignees

No one assigned

Labels

None yet

Milestone

No milestone