All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends
GpuIndex.cu
/**
* 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.
#include "GpuIndex.h"
#include "../FaissAssert.h"
#include "GpuResources.h"
#include "utils/DeviceUtils.h"
namespace faiss { namespace gpu {
GpuIndex::GpuIndex(GpuResources* resources,
int device,
int dims,
faiss::MetricType metric) :
Index(dims, metric),
resources_(resources),
device_(device) {
FAISS_ASSERT(device_ < getNumDevices());
FAISS_ASSERT(resources_);
resources_->initializeForDevice(device_);
}
} } // namespace
faiss::MetricType
MetricType
Some algorithms support both an inner product vetsion and a L2 search version.
Definition: Index.h:44