From 5501f7390a6d9821707cfa65254fb94f7c850086 Mon Sep 17 00:00:00 2001 From: Weiqun Zhang Date: Mon, 18 Mar 2024 17:51:17 -0700 Subject: [PATCH] Fix deprecation warning for CUDA 12.4 (#3811) `cudaDeviceSetSharedMemConfig` has been deprecated since CUDA 12.4. --- Src/Base/AMReX_GpuDevice.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Src/Base/AMReX_GpuDevice.cpp b/Src/Base/AMReX_GpuDevice.cpp index 2bdd6c87363..d699a20a2b5 100644 --- a/Src/Base/AMReX_GpuDevice.cpp +++ b/Src/Base/AMReX_GpuDevice.cpp @@ -374,11 +374,13 @@ Device::initialize_gpu () cudaDeviceGetAttribute(&memory_pools_supported, cudaDevAttrMemoryPoolsSupported, device_id); #endif +#if (__CUDACC_VER_MAJOR__ < 12) || ((__CUDACC_VER_MAJOR__ == 12) && (__CUDACC_VER_MINOR__ < 4)) if (sizeof(Real) == 8) { AMREX_CUDA_SAFE_CALL(cudaDeviceSetSharedMemConfig(cudaSharedMemBankSizeEightByte)); } else if (sizeof(Real) == 4) { AMREX_CUDA_SAFE_CALL(cudaDeviceSetSharedMemConfig(cudaSharedMemBankSizeFourByte)); } +#endif for (int i = 0; i < max_gpu_streams; ++i) { AMREX_CUDA_SAFE_CALL(cudaStreamCreate(&gpu_stream_pool[i]));