blob: e5e2910fe81066f12b2210789f9935e44598ff3c (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include <vespa/vespalib/datastore/atomic_entry_ref.h>
#include <vespa/vespalib/datastore/atomic_value_wrapper.h>
namespace search::tensor {
/**
* Represents a graph node for non-dense tensors (multiple nodes per document).
*/
class HnswNode {
using AtomicEntryRef = vespalib::datastore::AtomicEntryRef;
using EntryRef = vespalib::datastore::EntryRef;
AtomicEntryRef _levels_ref;
vespalib::datastore::AtomicValueWrapper<uint32_t> _docid;
vespalib::datastore::AtomicValueWrapper<uint32_t> _subspace;
public:
HnswNode() noexcept
: _levels_ref(),
_docid(),
_subspace()
{
}
AtomicEntryRef& levels_ref() noexcept { return _levels_ref; }
const AtomicEntryRef& levels_ref() const noexcept { return _levels_ref; }
void store_docid(uint32_t docid) noexcept { _docid.store_release(docid); }
void store_subspace(uint32_t subspace) noexcept { _subspace.store_release(subspace); }
// Mapping from nodeid to docid and subspace.
uint32_t acquire_docid() const noexcept { return _docid.load_acquire(); }
uint32_t acquire_subspace() const noexcept { return _subspace.load_acquire(); }
static constexpr bool identity_mapping = false;
};
}
|