cuda.hpp

/*
 * 版权所有 (c) 2022-2025, NVIDIA CORPORATION。
 *
 * 根据 Apache 许可证 2.0 版（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下位置获取许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则软件
 * 根据许可证分发，按“原样”提供，
 * 不附带任何明示或默示的担保或条件。
 * 请参阅许可证了解特定语言的管理权限和
 * 许可证下的限制。
 */
#pragma once

#include <kvikio/shim/cuda_h_wrapper.hpp>
#include <kvikio/shim/utils.hpp>

namespace kvikio {

class cudaAPI {
 public
 decltype(cuInit)* Init{nullptr};
 decltype(cuMemHostAlloc)* MemHostAlloc{nullptr};
 decltype(cuMemFreeHost)* MemFreeHost{nullptr};
 decltype(cuMemcpyHtoDAsync)* MemcpyHtoDAsync{nullptr};
 decltype(cuMemcpyDtoHAsync)* MemcpyDtoHAsync{nullptr};
 decltype(cuPointerGetAttribute)* PointerGetAttribute{nullptr};
 decltype(cuPointerGetAttributes)* PointerGetAttributes{nullptr};
 decltype(cuCtxPushCurrent)* CtxPushCurrent{nullptr};
 decltype(cuCtxPopCurrent)* CtxPopCurrent{nullptr};
 decltype(cuCtxGetCurrent)* CtxGetCurrent{nullptr};
 decltype(cuMemGetAddressRange)* MemGetAddressRange{nullptr};
 decltype(cuGetErrorName)* GetErrorName{nullptr};
 decltype(cuGetErrorString)* GetErrorString{nullptr};
 decltype(cuDeviceGet)* DeviceGet{nullptr};
 decltype(cuDevicePrimaryCtxRetain)* DevicePrimaryCtxRetain{nullptr};
 decltype(cuDevicePrimaryCtxRelease)* DevicePrimaryCtxRelease{nullptr};
 decltype(cuStreamSynchronize)* StreamSynchronize{nullptr};
 decltype(cuStreamCreate)* StreamCreate{nullptr};
 decltype(cuStreamDestroy)* StreamDestroy{nullptr};

 private
 cudaAPI();

 public
 cudaAPI(cudaAPI const&) = delete;
 void operator=(cudaAPI const&) = delete;

 KVIKIO_EXPORT static cudaAPI& instance();
};

#ifdef KVIKIO_CUDA_FOUND
bool is_cuda_available();
#else
constexpr bool is_cuda_available() { return false; }
#endif

} // 命名空间 kvikio