aligned.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020-2025, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <rmm/detail/export.hpp>
 
#include <cassert>
#include <cstddef>
 
namespace RMM_EXPORT rmm {
 
static constexpr std::size_t CUDA_ALLOCATION_ALIGNMENT{256};
 
[[nodiscard]] bool is_pow2(std::size_t value) noexcept;
 
[[nodiscard]] bool is_supported_alignment(std::size_t alignment) noexcept;
 
[[nodiscard]] std::size_t align_up(std::size_t value, std::size_t alignment) noexcept;
 
[[nodiscard]] std::size_t align_down(std::size_t value, std::size_t alignment) noexcept;
 
[[nodiscard]] bool is_aligned(std::size_t value, std::size_t alignment) noexcept;
 
[[nodiscard]] bool is_pointer_aligned(void* ptr,
                                      std::size_t alignment = CUDA_ALLOCATION_ALIGNMENT) noexcept;
  // end of group
 
}  // namespace RMM_EXPORT rmm