detail/utils.hpp

/*
 * SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
#pragma once
 
#include <cstring>
#include <exception>
#include <utility>
 
#include <kvikio/error.hpp>
 
namespace kvikio::detail {
 
[[nodiscard]] std::size_t align_up(std::size_t value, std::size_t alignment);
 
[[nodiscard]] void* align_up(void* addr, std::size_t alignment);
 
[[nodiscard]] std::size_t align_down(std::size_t value, std::size_t alignment);
 
[[nodiscard]] void* align_down(void* addr, std::size_t alignment);
 
bool is_aligned(std::size_t value, std::size_t alignment);
 
bool is_aligned(void* addr, std::size_t alignment);
 
template <typename F>
class ScopeExit {
  static_assert(std::is_invocable_v<F>, "ScopeExit callable must be invocable with no arguments");
 
 private:
  F _cleanup;
 
 public:
  [[nodiscard]] explicit ScopeExit(F&& cleanup) : _cleanup(std::move(cleanup)) {}
 
  ~ScopeExit() noexcept
  {
    try {
      _cleanup();
    } catch (std::exception const& e) {
      KVIKIO_LOG_ERROR(e.what());
    } catch (...) {
      KVIKIO_LOG_ERROR("Unhandled exception");
    }
  }
 
  ScopeExit(ScopeExit const&)            = delete;
  ScopeExit& operator=(ScopeExit const&) = delete;
  ScopeExit(ScopeExit&&)                 = delete;
  ScopeExit& operator=(ScopeExit&&)      = delete;
};
 
}  // namespace kvikio::detail