scoped_memory_record.hpp

 
#pragma once
 
#include <cstdint>
#include <type_traits>
 
namespace rapidsmpf {
 
struct ScopedMemoryRecord {
    [[nodiscard]] std::int64_t num_total_allocs() const noexcept;
 
    [[nodiscard]] std::int64_t num_current_allocs() const noexcept;
 
    [[nodiscard]] std::int64_t current() const noexcept;
 
    [[nodiscard]] std::int64_t total() const noexcept;
 
    [[nodiscard]] std::int64_t peak() const noexcept;
 
    [[nodiscard]] std::int64_t max() const noexcept;
 
    void record_allocation(std::int64_t nbytes);
 
    void record_deallocation(std::int64_t nbytes);
 
    ScopedMemoryRecord& add_subscope(ScopedMemoryRecord const& subscope);
 
    ScopedMemoryRecord& add_scope(ScopedMemoryRecord const& scope);
 
  private:
    std::int64_t num_current_allocs_{0};
    std::int64_t num_total_allocs_{0};
    std::int64_t current_{0};
    std::int64_t total_{0};
    std::int64_t peak_{0};
    std::int64_t max_{0};
};
 
static_assert(
    std::is_trivially_copyable_v<ScopedMemoryRecord>,
    "ScopedMemoryRecord must be trivially copyable"
);
 
}  // namespace rapidsmpf