mirror of
https://github.com/facebookresearch/faiss.git
synced 2025-06-03 21:54:02 +08:00
784e2facd8
* moved most FAISS_ASSERT calls to C++ exceptions, and adjusted memory allocation to avoid mem leaks * added an IndexIVFScalarQuantizer type that offers an intermediate compression between IVFFlat and IVFPQ * support removal of indices in IndexIDMap / IndexFlat combination * various fixes in GPU code
27 lines
572 B
Plaintext
27 lines
572 B
Plaintext
/**
|
|
* Copyright (c) 2015-present, Facebook, Inc.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the CC-by-NC license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
// Copyright 2004-present Facebook. All Rights Reserved.
|
|
#pragma once
|
|
|
|
namespace faiss { namespace gpu {
|
|
|
|
template <typename T>
|
|
inline __device__ void swap(bool swap, T& x, T& y) {
|
|
T tmp = x;
|
|
x = swap ? y : x;
|
|
y = swap ? tmp : y;
|
|
}
|
|
|
|
template <typename T>
|
|
inline __device__ void assign(bool assign, T& x, T y) {
|
|
x = assign ? y : x;
|
|
}
|
|
|
|
} } // namespace
|