Skip to content

Enable memory-aware row tiling for GPU CD consensus refits - #10

Draft
JerryIshihara wants to merge 2 commits into
TrynkaLab:mainfrom
JerryIshihara:features/nmf-gpu-kernal-tiling
Draft

Enable memory-aware row tiling for GPU CD consensus refits#10
JerryIshihara wants to merge 2 commits into
TrynkaLab:mainfrom
JerryIshihara:features/nmf-gpu-kernal-tiling

Conversation

@JerryIshihara

Copy link
Copy Markdown

Summary

Adds row-wise tiling to the GPU coordinate-descent solver for the fixed-H refits performed during cNMF consensus.

This reduces peak GPU memory usage on large datasets while preserving CD numerical behavior and the existing consensus workflow.

Changes

  • Add tiled GPU CD execution for update_H=False.
  • Keep consensus spectra H fixed while updating usage matrix W in row batches.
  • Reuse shared fixed-H products across tiles.
  • Automatically select the row-batch size from available GPU VRAM.
  • Add manual control with: --gpu-row-tiling-ratio 0.5

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant