xla
[Pallas] Make repeat_with_fixed_output_size not OOM on VMEM
#7145

Merged

[Pallas] Make repeat_with_fixed_output_size not OOM on VMEM #7145

alanwaketan merged 4 commits into master from alanwaketan/tgmm2

initial commit

d428dea7

fix linters

faa9f099

alanwaketan requested a review from

miladm 1 year ago

alanwaketan requested a review from

wonjoo-wj 1 year ago

alanwaketan requested a review from

JackCaoG 1 year ago

JackCaoG approved these changes on 2024-05-29

alanwaketan added tpuci

Add a check

b6e9e8e5

JackCaoG approved these changes on 2024-05-29

Separate cumsum out

4e708380

alanwaketan merged ce1205e1 into master 1 year ago

alanwaketan deleted the alanwaketan/tgmm2 branch 1 year ago

Reviewers

JackCaoG

miladm

wonjoo-wj

Assignees

No one assigned

Labels

None yet

Milestone

No milestone