blob: dfa5e60f19c4f325993acbd76d82b1cee100dc2d (
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include "attribute_transient_memory_calculator.h"
#include <vespa/searchlib/attribute/attribute_header.h>
#include <vespa/searchcommon/attribute/config.h>
#include <vespa/searchlib/attribute/loadedenumvalue.h>
#include <vespa/searchlib/attribute/loadedvalue.h>
#include <vespa/searchlib/attribute/attributevector.h>
using search::attribute::BasicType;
namespace proton {
namespace {
size_t
get_transient_memory_usage(bool old_enumerated,
const search::attribute::Config& new_config,
uint64_t total_value_count)
{
if (new_config.fastSearch()) {
if (old_enumerated) {
return sizeof(search::attribute::LoadedEnumAttribute) * total_value_count;
} else {
switch (new_config.basicType().type()) {
case BasicType::Type::INT8:
return sizeof(search::attribute::LoadedValue<int8_t>) * total_value_count;
case BasicType::Type::INT16:
return sizeof(search::attribute::LoadedValue<int16_t>) * total_value_count;
case BasicType::Type::INT32:
return sizeof(search::attribute::LoadedValue<int32_t>) * total_value_count;
case BasicType::Type::INT64:
return sizeof(search::attribute::LoadedValue<int64_t>) * total_value_count;
case BasicType::Type::FLOAT:
return sizeof(search::attribute::LoadedValue<float>) * total_value_count;
case BasicType::Type::DOUBLE:
return sizeof(search::attribute::LoadedValue<double>) * total_value_count;
default:
;
}
}
}
return 0u;
}
}
size_t
AttributeTransientMemoryCalculator::operator()(const search::AttributeVector& attribute_vector,
const search::attribute::Config& new_config) const
{
uint64_t total_value_count = attribute_vector.getStatus().getNumValues();
bool old_enumerated = attribute_vector.getEnumeratedSave();
return get_transient_memory_usage(old_enumerated, new_config, total_value_count);
}
size_t
AttributeTransientMemoryCalculator::operator()(const search::attribute::AttributeHeader& old_header,
const search::attribute::Config& new_config) const
{
return get_transient_memory_usage(old_header.getEnumerated(), new_config, old_header.get_total_value_count());
};
}
|
