xla

83dc9da8 - Calculate vmem limit dynamically in the quantized matmul kernel. (#9470)