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
|
#ifndef STRUCTURES_RESOURCES_H_
#define STRUCTURES_RESOURCES_H_
#include <cstdint>
#include <cstddef>
namespace wsclean {
class Resources {
public:
Resources() = default;
Resources(std::size_t n_cpus, int64_t memory)
: n_cpus_(n_cpus), memory_(memory) {}
size_t NCpus() const { return n_cpus_; }
int64_t Memory() const { return memory_; }
Resources GetCombined(size_t num_combined) const;
Resources GetPart(size_t part_size) const;
private:
std::size_t n_cpus_;
int64_t memory_;
};
/**
* Obtain available system memory, taking into account user-requested limits on
* the usage. If both memory_fraction and abs_memory_limit reduce the amount of
* available memory, they're combined by taking the strongest limit of the two.
* They're not applied cummulative.
* @param memory_fraction Value (0-1) to limit relative memory.
* @param abs_memory_limit Absolute limit in gb, or 0 if no absolute limit is
* set.
* @returns available memory in bytes.
*/
int64_t GetAvailableMemory(double memory_fraction, double abs_memory_limit);
} // namespace wsclean
#endif
|
