Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorTopK.cu617 int max_blocks_per_mp = 32; in get_items_per_thread() local
621 int max_blocks_per_mp = prop->maxBlocksPerMultiProcessor; in get_items_per_thread() local
623 int max_blocks_per_mp = 32; in get_items_per_thread() local