af874bf1e5
On AMD/ROCm a clean launch stalls at "Requested to load LTXAV" while system RAM fills and spills to swap. This is host-side pinned-memory exhaustion, not VRAM pressure. The static pin_memory path ignores ensure_pin_registerable()'s result and does not check ensure_pin_budget() before cudaHostRegister. It can therefore keep registering offloaded weights after current host-RAM pressure says to stop. Honor both existing helpers before registering. If either budget cannot be met, leave the tensor in pageable RAM. Add focused coverage for RAM-budget rejection, registration-budget rejection, successful accounting, and the --high-ram contract. |
||
|---|---|---|
| .. | ||
| app_test | ||
| assets_test | ||
| comfy_api_test | ||
| comfy_extras_test | ||
| comfy_quant | ||
| comfy_test | ||
| execution_test | ||
| folder_paths_test | ||
| jobs_cancel_test | ||
| prompt_server_test | ||
| security_test | ||
| seeder_test | ||
| server/utils | ||
| server_test | ||
| utils | ||
| README.md | ||
| deploy_environment_test.py | ||
| feature_flags_test.py | ||
| requirements.txt | ||
| websocket_feature_flags_test.py | ||
README.md
Pytest Unit Tests
Install test dependencies
pip install -r tests-unit/requirements.txt
Run tests
pytest tests-unit/