ENH: Expose flag to release GPU memory when dirty

RTKConsortium · Sep 6, 2024 · 7a76f08 · 7a76f08
1 parent e2f19a1
commit 7a76f08
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/include/itkCudaDataManager.h b/include/itkCudaDataManager.h
@@ -162,6 +162,11 @@ class CudaCommon_EXPORT CudaDataManager : public Object
   void
   SetGPUDirtyFlag(bool isDirty);
 
+  /** Controls whether GPU memory should be released when dirty. On by default.
+   * When turning it off, one must call Free() to release the GPU memory. */
+  void
+  SetReleaseDirtyGPUBufferFlag(bool release);
+
   /** Make GPU up-to-date and mark CPU as dirty.
    * Call this function when you want to modify CPU data */
   void

diff --git a/src/itkCudaDataManager.cxx b/src/itkCudaDataManager.cxx
@@ -124,6 +124,12 @@ CudaDataManager::SetGPUDirtyFlag(bool isDirty)
     this->Free();
 }
 
+void
+CudaDataManager::SetReleaseDirtyGPUBufferFlag(bool release)
+{
+  m_ReleaseDirtyGPUBuffer = release;
+}
+
 void
 CudaDataManager::SetGPUBufferDirty()
 {