Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dgeneral_reduction_impl.cu61 S threadV = init_V; in ThreadReduction() local
93 S threadV = init_V; in WarpReduction() local
146 S threadV = init_V; in Warp4Reduction() local
234 S threadV = init_V; in BlockReduction() local
Dtopk_impl.cu102 … S *shared_V, int *watermark, T *threadK, S *threadV, T *ceil_K, S *ceil_V, S *k_prime) { in TopKStep()
189 S threadV[thread_queue]; // NOLINT in TopKBlock() local
Drcwm_small_impl.cu55 S threadV[thread_queue]; // NOLINT in L2Rcwm() local
Dtopk_lib.cuh460 inline __device__ void MergeWarpQueue(T *threadK, S *threadV, T *warp_K, S *warp_V) { in MergeWarpQueue()