mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
21 lines
441 B
Plaintext
21 lines
441 B
Plaintext
|
/**
|
||
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
||
|
|
* All rights reserved.
|
||
|
|
*
|
||
|
|
* This source code is licensed under the BSD+Patents license found in the
|
||
|
|
* LICENSE file in the root directory of this source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include "BlockSelectImpl.cuh"
|
||
|
|
#include "../DeviceDefs.cuh"
|
||
|
|
|
||
|
|
namespace faiss { namespace gpu {
|
||
|
|
|
||
|
|
#if GPU_MAX_SELECTION_K >= 2048
|
||
|
|
#ifdef FAISS_USE_FLOAT16
|
||
|
|
BLOCK_SELECT_IMPL(half, true, 2048, 8);
|
||
|
|
#endif
|
||
|
|
#endif
|
||
|
|
|
||
|
|
} } // namespace
|