Skip to content

Add GPU fixed-H consensus refit#4

Merged
OlivierBakker merged 1 commit into
TrynkaLab:mainfrom
JerryIshihara:features/nmf-gpu-kernal-wiring-consensus
Jul 1, 2026
Merged

Add GPU fixed-H consensus refit#4
OlivierBakker merged 1 commit into
TrynkaLab:mainfrom
JerryIshihara:features/nmf-gpu-kernal-wiring-consensus

Conversation

@JerryIshihara

Copy link
Copy Markdown

Summary

Adds GPU support for the fixed-H NMF refit path used during cNMF consensus.

Changes

  • Adds a fixed-H multiplicative-update path for update_H=False
  • Keeps fixed spectra H unchanged and updates only usages W
  • Adds validation for fixed H shape, finiteness, and non-negativity
  • Splits normal NMF and fixed-H refit paths into separate helpers for readability
  • Allows --engine gpu for both factorize and consensus
  • Keeps the standalone GPU kernel API unchanged

@OlivierBakker OlivierBakker merged commit 4d9f837 into TrynkaLab:main Jul 1, 2026
1 check passed
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.

2 participants