xla
Fallback to CPU for dot implementation for long tensors
#6708

Merged

Fallback to CPU for dot implementation for long tensors #6708

bhavya01 merged 9 commits into master from long_dot

Fallback to CPU for dot implementation for long tensors

58fcee07

fix cpp test

5cf5428f

bhavya01 assigned

bhavya01 1 year ago

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

bhavya01 requested a review from

JackCaoG 1 year ago

JackCaoG commented on 2024-03-11

Merge branch 'master' into long_dot

3b3e966d

minor fixes

dae445a5

miladm requested a review from

miladm 1 year ago

Only fallback to CPU for TPU device

183e2ea7

fix test

aee8b1a5

Fix dot implementation for all integral types

919ef97f

Fix lint

c7c25eec

bhavya01 requested a review from

JackCaoG 1 year ago

JackCaoG commented on 2024-03-13

JackCaoG approved these changes on 2024-03-18

Add a comment that int64 and uint64 dot products are not supported fo…

7fa889e9

bhavya01 merged 7145a719 into master 1 year ago

bhavya01 deleted the long_dot branch 1 year ago

Reviewers

JackCaoG

miladm

Assignees

bhavya01

Labels

None yet

Milestone

No milestone