Faiss
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
faiss::gpu::BlockSelect< K, V, Dir, Comp, 1, NumThreadQ, ThreadsPerBlock > Struct Template Reference

Specialization for k == 1 (NumWarpQ == 1) More...

Public Member Functions

__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 ()
 

Public Attributes

threadK
 
threadV
 
K * sharedK
 
V * sharedV
 

Static Public Attributes

static constexpr int kNumWarps = ThreadsPerBlock / kWarpSize
 

Detailed Description

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 273 of file Select.cuh.


The documentation for this struct was generated from the following file: