aligned.hpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <rmm/detail/export.hpp>
 
#include <cassert>
#include <cstddef>
 
RMM_NAMESPACE_BEGIN


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]] bool is_supported_base_resource_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 = rmm::CUDA_ALLOCATION_ALIGNMENT) noexcept;
  // end of group
 
RMM_NAMESPACE_END