mirror of
https://github.com/Comfy-Org/comfy-kitchen.git
synced 2026-11-04 23:37:42 +00:00
[PR #281] Add memmove kernel (For linear combiner) #281
Labels
No labels
pull-request
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
Comfy-Org/comfy-kitchen#281
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/Comfy-Org/comfy-kitchen/pull/281
Author: @rattus128
Created: 10/26/2026
Status: 🔄 Open
Base:
main← Head:prs/memmove-for-linear-merge📝 Commits (2)
3f46c9bAdd memmove for overlapping device byte ranges7377a05tests: memmove overlap cases📊 Changes
10 files changed (+261 additions, -0 deletions)
View changed files
📝
comfy_kitchen/__init__.py(+10 -0)📝
comfy_kitchen/backends/cuda/CMakeLists.txt(+1 -0)📝
comfy_kitchen/backends/cuda/__init__.py(+9 -0)📝
comfy_kitchen/backends/cuda/dlpack_bindings.cpp(+20 -0)➕
comfy_kitchen/backends/cuda/ops/memmove.cu(+80 -0)📝
comfy_kitchen/backends/hip/CMakeLists.txt(+1 -0)📝
comfy_kitchen/backends/hip/__init__.py(+6 -0)📝
comfy_kitchen/backends/hip/dlpack_bindings.cpp(+16 -0)➕
comfy_kitchen/backends/hip/ops/memmove.hip(+80 -0)➕
tests/test_memmove.py(+38 -0)📄 Description
Add a kernel for in-place memmove of tensors. Useful for relaying out tensors already in VRAM without causing a VRAM allocation spike. Used by ComfyUI linear combiner.
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.