Specialization for k == 1 (NumWarpQ == 1)
More...
|
__device__ | BlockSelect (K initK, V initV, K *smemK, V *smemV, int k) |
|
__device__ void | addThreadQ (K k, V v) |
|
__device__ void | checkThreadQ () |
|
__device__ void | add (K k, V v) |
|
__device__ void | reduce () |
|
|
K | threadK |
|
V | threadV |
|
K * | sharedK |
|
V * | sharedV |
|
|
static constexpr int | kNumWarps = ThreadsPerBlock / kWarpSize |
|
template<typename K, typename V, bool Dir, typename Comp, int NumThreadQ, int ThreadsPerBlock>
struct faiss::gpu::BlockSelect< K, V, Dir, Comp, 1, NumThreadQ, ThreadsPerBlock >
Specialization for k == 1 (NumWarpQ == 1)
Definition at line 267 of file Select.cuh.
The documentation for this struct was generated from the following file:
- /data/users/matthijs/github_faiss/faiss/gpu/utils/Select.cuh