Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dflash_fwd_kernel.h747 const int block_table_idx_cur = n_block * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
843 const int block_table_idx_cur = (n_block + 1) * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
878 const int block_table_idx_cur = n_block * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
921 const int block_table_idx_cur = (n_block + 1) * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local
942 const int block_table_idx_cur = n_block * kBlockN / params.page_block_size; in compute_attn_1rowblock_splitkv() local