xla

9ec626e6 - Use pages_per_seq * page_size instead of directly passing max_model_len (#8950)