Skip to content

Free GPU memory eagerly#689

Open
jpsamaroo wants to merge 9 commits intomasterfrom
jps/lu-ldiv-free
Open

Free GPU memory eagerly#689
jpsamaroo wants to merge 9 commits intomasterfrom
jps/lu-ldiv-free

Conversation

@jpsamaroo
Copy link
Member

@jpsamaroo jpsamaroo commented Feb 27, 2026

We now use unsafe_free! to eagerly free Datadeps and DArray GPU allocations when safe to do so.

This PR also:

  • Fixes LU for GPUs
  • Fixes Cholesky for DVector inputs
  • Adds + and - methods for DArrays
  • Fixes a scheduler bug around error reporting in large task graphs
  • Adapts RefValue to a new GPURef, required for LU and other algorithms

Todo:

  • Add unsafe_free! to stencils
  • Evaluate if QR/CAQR can use unsafe_free!
  • Add CPU/GPU tests for Cholesky with DVector inputs
  • Add GPU tests for Cholesky and LU
  • Add GPU tests for + and -
  • Add BLAS alternatives for GPU backends which don't have BLAS wired up

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant