Fic incorrect include in cuda_pinned_memory_resource.h

We need `cuda_runtime.h`for the C++ `cudaMallocHost` and not `cuda_runtime_api.h`
NVIDIA · Apr 17, 2024 · 0330422 · 0330422
1 parent 9a9bffd
commit 0330422
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libcudacxx/include/cuda/__memory_resource/cuda_pinned_memory_resource.h b/libcudacxx/include/cuda/__memory_resource/cuda_pinned_memory_resource.h
@@ -24,15 +24,15 @@
 #if !defined(_CCCL_COMPILER_MSVC_2017)
 
 #  if !defined(_CCCL_CUDA_COMPILER_NVCC) && !defined(_CCCL_CUDA_COMPILER_NVHPC)
-#    include <cuda_runtime_api.h>
+#    include <cuda_runtime.h>
 #  endif // !_CCCL_CUDA_COMPILER_NVCC && !_CCCL_CUDA_COMPILER_NVHPC
 
 #  include <cuda/__memory_resource/get_property.h>
 #  include <cuda/__memory_resource/properties.h>
 #  include <cuda/__memory_resource/resource.h>
 #  include <cuda/__memory_resource/resource_ref.h>
 #  include <cuda/std/__cuda/api_wrapper.h>
-#  include <cuda/std/detail/libcxx/include/__new/bad_alloc.h>
+#  include <cuda/std/__new/bad_alloc.h>
 
 #  if _CCCL_STD_VER >= 2014