From b5ca920dee762ebf1d25e8562ab07ce725fcc55a Mon Sep 17 00:00:00 2001 From: "Betigeri, Sourabh" Date: Fri, 15 Aug 2025 22:44:41 -0700 Subject: [PATCH] =?UTF-8?q?SWDEV-546293=20-=20Adds=20nvidia=20mappings=20t?= =?UTF-8?q?o=20hipMemPrefetchAsync=5Fvs=20and=20hip=E2=80=A6=20(#36)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit SWDEV-546293 - Adds nvidia mappings to hipMemPrefetchAsync_vs and hipMemAdvise_v2 [ROCm/hipother commit: bf0302a4944319450e1a26646d82581ff08a327b] --- .../nvidia_detail/nvidia_hip_runtime_api.h | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h index 047538fab1..5effd3fa70 100644 --- a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h +++ b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h @@ -2050,6 +2050,27 @@ inline static hipError_t hipMemPrefetchAsync(const void* dev_ptr, size_t count, return hipCUDAErrorTohipError(cudaMemPrefetchAsync(dev_ptr, count, device, stream)); } +inline static hipError_t hipMemPrefetchAsync_v2(const void *dev_ptr, size_t count, + hipMemLocation location, unsigned int flags, + hipStream_t stream __dparm(0)) { +#if CUDA_VERSION >= 13000 + return hipCUDAErrorTohipError(cudaMemPrefetchAsync(dev_ptr, count, location, flags, stream)); +#else + return hipCUDAErrorTohipError(cudaMemPrefetchAsync_v2(dev_ptr, count, location, flags, + stream)); +#endif +} + +inline static hipError_t hipMemAdvise_v2(const void* dev_ptr, size_t count, + hipMemoryAdvise advice, hipMemLocation location) { +#if CUDA_VERSION >= 13000 + return hipCUDAErrorTohipError(cudaMemAdvise(dev_ptr, count, advice , location)); +#else + return hipCUDAErrorTohipError(cudaMemAdvise_v2(dev_ptr, count, advice , location)); +#endif +} + + inline static hipError_t hipMemRangeGetAttribute(void* data, size_t data_size, hipMemRangeAttribute attribute, const void* dev_ptr, size_t count) {