xla

25becbf2 - Use pages_per_seq * page_size instead of directly passing max_model_len