BUG: Remove CudaContextManager class and use cudaSetDevice

Use the primary context with cudaSetDevice() introduced by Cuda 7 (https://developer.download.nvidia.com/compute/cuda/7_0/Prod/doc/CUDA_Toolkit_Release_Notes.pdf) instead of a new one. cudaSetDevice is called before every memory transfer between the CPU and GPU to be sure that the context is set for the current thread, see https://developer.nvidia.com/blog/cuda-pro-tip-always-set-current-device-avoid-multithreading-bugs/.

include/itkCudaDataManager.h

@@ -22,7 +22,6 @@
 #include "itkDataObject.h"
 #include "itkObjectFactory.h"
 #include "itkCudaUtil.h"
-#include "itkCudaContextManager.h"
 #include "CudaCommonExport.h"
 
 #include <mutex>
@@ -232,6 +231,8 @@ class CudaCommon_EXPORT CudaDataManager : public Object
   void
   PrintSelf(std::ostream & os, Indent indent) const override;
 
+  int m_Device;
+
 private:
   CudaDataManager(const Self &) = delete; // purposely not implemented
   void
@@ -240,8 +241,6 @@ class CudaCommon_EXPORT CudaDataManager : public Object
 protected:
   size_t m_BufferSize; // # of bytes
 
-  CudaContextManager * m_ContextManager;
-
   /** buffer type */
   int m_MemFlags;
 

include/itkCudaImageDataManager.h

@@ -23,7 +23,6 @@
 #include <itkObjectFactory.h>
 #include "itkCudaUtil.h"
 #include "itkCudaDataManager.h"
-#include "itkCudaContextManager.h"
 
 namespace itk
 {

include/itkCudaImageDataManager.hxx

@@ -76,9 +76,7 @@ CudaImageDataManager<ImageType>::MakeCPUBufferUpToDate()
       std::cout << this << ": GPU->CPU data copy" << std::endl;
 #endif
 
-      CUDA_CHECK(cuCtxSetCurrent(
-        *(this->m_ContextManager->GetCurrentContext()))); // This is necessary when running multithread to bind the host
-                                                          // CPU thread to the right context
+      CUDA_CHECK(cudaSetDevice(m_Device));
       errid = cudaMemcpy(m_CPUBuffer, m_GPUBuffer->GetPointer(), m_BufferSize, cudaMemcpyDeviceToHost);
       CudaCheckError(errid, __FILE__, __LINE__, ITK_LOCATION);
 
@@ -117,9 +115,7 @@ CudaImageDataManager<ImageType>::MakeGPUBufferUpToDate()
       std::cout << "CPU->GPU data copy" << std::endl;
 #endif
 
-      CUDA_CHECK(cuCtxSetCurrent(
-        *(this->m_ContextManager->GetCurrentContext()))); // This is necessary when running multithread to bind the host
-                                                          // CPU thread to the right context
+      CUDA_CHECK(cudaSetDevice(m_Device));
       errid = cudaMemcpy(m_GPUBuffer->GetPointer(), m_CPUBuffer, m_BufferSize, cudaMemcpyHostToDevice);
       CudaCheckError(errid, __FILE__, __LINE__, ITK_LOCATION);
 

src/CMakeLists.txt

@@ -1,5 +1,4 @@
 set(CudaCommon_SRCS
-  itkCudaContextManager.cxx
   itkCudaDataManager.cxx
   itkCudaUtil.cxx
   itkCudaMemoryProbe.cxx

src/itkCudaDataManager.cxx

@@ -24,18 +24,8 @@ namespace itk
 // constructor
 CudaDataManager::CudaDataManager()
 {
-  m_ContextManager = CudaContextManager::GetInstance();
-
-  // Creating the context in the constructor allows avoiding a memory leak.
-  // However, the cuda data manager is created even if there is no use of CUDA
-  // software and sometimes one compiles RTK with CUDA but wants to use it
-  // without CUDA. So if the context pointer is nullptr, which indicates that there
-  // is no CUDA device available, we just do not set the context (SR). This fixes
-  // the problem reported here:
-  // https://www.creatis.insa-lyon.fr/pipermail/rtk-users/2015-July/000570.html
-  CUcontext * ctx = m_ContextManager->GetCurrentContext();
-  if (ctx)
-    CUDA_CHECK(cuCtxSetCurrent(*ctx));
+  m_Device = 0;
+  CUDA_CHECK(cudaSetDevice(m_Device));
 
   m_CPUBuffer = nullptr;
   m_GPUBuffer = GPUMemPointer::New();
@@ -56,7 +46,6 @@ CudaDataManager::CudaDataManager()
 CudaDataManager::~CudaDataManager()
 {
   m_GPUBuffer = nullptr;
-  CudaContextManager::DestroyInstance();
 }
 
 void
@@ -91,9 +80,7 @@ CudaDataManager::Free()
   {
     try
     {
-      CUDA_CHECK(cuCtxSetCurrent(
-        *(this->m_ContextManager->GetCurrentContext()))); // This is necessary when running multithread to bind the host
-                                                          // CPU thread to the right context
+      CUDA_CHECK(cudaSetDevice(m_Device));
       m_GPUBuffer->Free();
     }
     catch (itk::ExceptionObject & e)
@@ -171,9 +158,7 @@ CudaDataManager::UpdateCPUBuffer()
       std::cout << this << "::UpdateCPUBuffer GPU->CPU data copy " << m_GPUBuffer->GetPointer() << "->" << m_CPUBuffer
                 << " : " << m_BufferSize << std::endl;
 #endif
-      CUDA_CHECK(cuCtxSetCurrent(
-        *(this->m_ContextManager->GetCurrentContext()))); // This is necessary when running multithread to bind the host
-                                                          // CPU thread to the right context
+      CUDA_CHECK(cudaSetDevice(m_Device));
       CUDA_CHECK(cudaMemcpy(m_CPUBuffer, m_GPUBuffer->GetPointer(), m_BufferSize, cudaMemcpyDeviceToHost));
       m_IsCPUBufferDirty = false;
     }
@@ -212,9 +197,7 @@ CudaDataManager::UpdateGPUBuffer()
       std::cout << this << "::UpdateGPUBuffer CPU->GPU data copy " << m_CPUBuffer << "->" << m_GPUBuffer->GetPointer()
                 << " : " << m_BufferSize << std::endl;
 #endif
-      CUDA_CHECK(cuCtxSetCurrent(
-        *(this->m_ContextManager->GetCurrentContext()))); // This is necessary when running multithread to bind the host
-                                                          // CPU thread to the right context
+      CUDA_CHECK(cudaSetDevice(m_Device));
       CUDA_CHECK(cudaMemcpy(m_GPUBuffer->GetPointer(), m_CPUBuffer, m_BufferSize, cudaMemcpyHostToDevice));
     }
     m_IsGPUBufferDirty = false;
@@ -259,7 +242,6 @@ CudaDataManager::Graft(const CudaDataManager * data)
   if (data)
   {
     m_BufferSize = data->m_BufferSize;
-    m_ContextManager = data->m_ContextManager;
     m_GPUBuffer = data->m_GPUBuffer;
     m_CPUBuffer = data->m_CPUBuffer;
     m_IsCPUBufferDirty = data->m_IsCPUBufferDirty;