WarpSelectHalfT512.cu 371 字节
Newer Older
J
JinHai-CN 已提交
1 2 3 4 5 6 7 8 9 10 11
/**
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#include <faiss/gpu/utils/warpselect/WarpSelectImpl.cuh>

namespace faiss { namespace gpu {

S
shengjun.li 已提交
12
#ifdef FAISS_USE_FLOAT16
J
JinHai-CN 已提交
13
WARP_SELECT_IMPL(half, true, 512, 8);
S
shengjun.li 已提交
14
#endif
J
JinHai-CN 已提交
15 16

} } // namespace