Skip to content

fix harmony_integrate TypeError under RMM managed memory - #764

Closed
A-Jolly-Holly wants to merge 4 commits into
scverse:mainfrom
A-Jolly-Holly:harmony-managed-empty-placeholder-bugfix
Closed

fix harmony_integrate TypeError under RMM managed memory#764
A-Jolly-Holly wants to merge 4 commits into
scverse:mainfrom
A-Jolly-Holly:harmony-managed-empty-placeholder-bugfix

Conversation

@A-Jolly-Holly

Copy link
Copy Markdown

Fixes #763
With a single batch key, the zero-size joint placeholder arrays own no allocation, so they report kDLCUDA while every other argument under rmm managed memory reports kDLCUDAManaged. No clustering_loop overload matches.
Using a zero-length view of cats keeps the memory kind consistent under any allocator. Verified on 0.16.1 hardware by applying the change to the installed package and rerunning the repro from 763
Added test_managed_memory.py

@A-Jolly-Holly A-Jolly-Holly changed the title Harmony managed empty placeholder bugfix fix harmony_integrate TypeError under RMM managed memory Aug 15, 2026
@Intron7

Intron7 commented Aug 19, 2026

Copy link
Copy Markdown
Member

@AlexHolly-AllenOps thank you for reporting this. Your fix is technically correct however I relies on a view not beeing looked at. So i made those arrays needed optional. I continued working on this in #766. Youre still credited and also i worked of your branch

@Intron7 Intron7 closed this Aug 19, 2026
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.

[BUG]harmony_integrate TypeError with rmm managed memory (0.16.1)

2 participants