paged_attention results in acore dumpedon TPU-v3#23825 Open OhadRubinopened this issueSep 22, 2024· 1 comment Description code to reproduce: if__name__=='__main__':# Set up example inputsbatch_size=8num_pages=32page_size=16head_dim=128total_num_pages=64*8num_kv_heads=num_heads=4...