cuda_utils.hpp

/*
 * 版权所有 (c) 2022-2024, NVIDIA CORPORATION。
 *
 * 获得 Apache License, Version 2.0 (“许可证”) 的许可；
 * 除非符合许可证，否则您不得使用此文件。
 * 您可以在以下网址获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 在许可证下分发，按“原样”基础提供，
 * 不提供任何明示或暗示的担保或条件。
 * 有关管理许可和
 * 限制的特定语言，请参见许可证。
 */
 
#pragma once
#include <utility>
 
#ifdef __CUDACC__
#define CUSPATIAL_HOST_DEVICE __host__ __device__
#define CUSPATIAL_KERNEL __global__ static
#else
#define CUSPATIAL_HOST_DEVICE
#define CUSPATIAL_KERNEL
#endif

template <std::size_t threads_per_block = 256>
std::pair<std::size_t, std::size_t> constexpr grid_1d(std::size_t const n)
{
 return {threads_per_block, (n + threads_per_block - 1) / threads_per_block};
}