Home
last modified time | relevance | path

Searched defs:block_table_idx_next (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dflash_fwd_kernel.h749 … const int block_table_idx_next = (n_block - 1) * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
845 const int block_table_idx_next = n_block * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
880 const int block_table_idx_next = (n_block - 1) * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
923 const int block_table_idx_next = n_block * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
944 const int block_table_idx_next = (n_block - 1) * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local