Comfy-Org / Comfy-Org/ComfyUI-Manager

Bug Report: RTX 5090 incompatible with default Pinned Memory

Open
#2,290 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
16.1k
Forks
2.5k
Avg merge
5d 4h
Merged PRs (30d)
13

Description

## Bug Report: RTX 5090 incompatible with default Pinned Memory

**Environment:**
- GPU: NVIDIA GeForce RTX 5090 Laptop (24GB)
- OS: Windows
- CUDA: 12.4+
- PyTorch: 2.x

**Root Cause:**
The new default-enabled Pinned Memory calls `cudaHostRegister(ptr, size, 1)`
which returns "CUDA error: invalid argument" on Blackwell architecture GPUs.

**Fix Suggestion:**
1. Change `cudaHostRegister` flag from `1` to `0` (or use `cudaHostRegisterDefault`)
2. Add error handling to disable Pinned Memory if registration fails
3. Add compute capability check for Blackwell (>= 10.0)

**Workaround:**
Add `--disable-pinned-memory` to launch arguments.

Contributor guide

No contributing guide indexed for this repository

Research direction

Locate the pinned-memory implementation and the cudaHostRegister(ptr, size, 1) call; no file or test is named in the report. Reproduce on the listed RTX 5090 Windows environment, then evaluate the suggested flag change, registration-failure handling, and Blackwell compute-capability check. Done means pinned memory no longer causes the reported invalid-argument error, while the documented workaround remains available.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.