xla
Calculate vmem limit dynamically in the quantized matmul kernel.
#9470

Merged

Calculate vmem limit dynamically in the quantized matmul kernel. #9470

vanbasten23 merged 3 commits into master from xiowei/calculate_vmem_limit_dynamically

Set vmem_limit according to block sizes.

f30367ad

linter

66d6100d

vanbasten23 requested a review from

yaochengji 1 year ago

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

yaochengji approved these changes on 2025-07-10

yaochengji commented on 2025-07-10

make vmem_limit account for vreg spill etc.

2abb8568

vanbasten23 merged 83dc9da8 into master 1 year ago

Reviewers

yaochengji

Assignees

No one assigned

Labels

None yet

Milestone

No milestone