CMakeLists.txt

@@ -68,6 +68,10 @@ if (CUDA_ON_BACKEND STREQUAL "CUDA")
         	message("CUDA is compatible 11.2/3/4/5")
                 set(WARNING_SUPPRESSION_AND_OPTION_NVCC  -Xcudafe "--display_error_number --diag_suppress=20014 --diag_suppress=20013 --diag_suppress=20012 --diag_suppress=20011 --diag_suppress=611 --diag_suppress=550  --diag_suppress=186 --diag_suppress=128" --expt-extended-lambda)
                 set(WARNING_SUPPRESSION_AND_OPTION_NVCC_TEXT  "-Xcudafe \"--display_error_number --diag_suppress=20014 --diag_suppress=20013 --diag_suppress=20012 --diag_suppress=20011 --diag_suppress=611 --diag_suppress=550  --diag_suppress=186 --diag_suppress=128\" --expt-extended-lambda")
+	elseif ( CUDA_VERSION_MAJOR EQUAL 12 AND (CUDA_VERSION_MINOR EQUAL 0))
+        	message("CUDA is compatible 12")
+                set(WARNING_SUPPRESSION_AND_OPTION_NVCC  -Xcudafe "--display_error_number --diag_suppress=20014 --diag_suppress=20013 --diag_suppress=20012 --diag_suppress=20011 --diag_suppress=611 --diag_suppress=550  --diag_suppress=186 --diag_suppress=128" --expt-extended-lambda)
+                set(WARNING_SUPPRESSION_AND_OPTION_NVCC_TEXT  "-Xcudafe \"--display_error_number --diag_suppress=20014 --diag_suppress=20013 --diag_suppress=20012 --diag_suppress=20011 --diag_suppress=611 --diag_suppress=550  --diag_suppress=186 --diag_suppress=128\" --expt-extended-lambda")                
 	else()
 		message(FATAL_ERROR "CUDA is incompatible, version 9.2 10.1 10.2 11.1 11.2 11.3 11.4 is only supported")
 	endif()

src/memory/ExtPreAlloc.hpp

@@ -65,6 +65,27 @@ public:
 		mem.resize(size);
 	}
 
+	/*! \brief Set the internal memory if you did not do it in the constructor
+     *
+	 * \param size number of bytes
+	 * \param mem external memory, used if you want to keep the memory
+     *
+	 */
+	void setMemory(size_t size, Mem & mem)
+	{
+		this->mem = &mem;
+		mem.resize(size);
+	}
+
+	/*! \brief Get the internal memory if you did not do it in the constructor
+     *
+	 * \return the internal memory
+     *
+	 */
+	Mem * getMemory()
+	{
+		return this->mem;
+	}
 
 	/*! \brief Copy the memory from device to device
 	 *