ENH: Expose flag to release GPU memory when dirty

RTKConsortium · Sep 17, 2024 · 07b6a32 · 07b6a32
1 parent e1e5eb8
commit 07b6a32
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/include/itkCudaDataManager.h b/include/itkCudaDataManager.h
@@ -162,6 +162,12 @@ 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. */
+  itkGetConstMacro(ReleaseDirtyGPUBuffer, bool);
+  itkSetMacro(ReleaseDirtyGPUBuffer, bool);
+  itkBooleanMacro(ReleaseDirtyGPUBuffer);
+
   /** Make GPU up-to-date and mark CPU as dirty.
    * Call this function when you want to modify CPU data */
   void