Skip to content

Use PYTORCH_ALLOC_CONF instead of deprecated PYTORCH_CUDA_ALLOC_CONF#32

Merged
p-e-w merged 2 commits intop-e-w:masterfrom
kldzj:pytorch-deprecation-warning
Nov 21, 2025
Merged

Use PYTORCH_ALLOC_CONF instead of deprecated PYTORCH_CUDA_ALLOC_CONF#32
p-e-w merged 2 commits intop-e-w:masterfrom
kldzj:pytorch-deprecation-warning

Conversation

@kldzj
Copy link
Contributor

@kldzj kldzj commented Nov 20, 2025

Problem

With #17 I introduced setting PYTORCH_CUDA_ALLOC_CONF to expandable_segments:True to reduce memory fragmentation for the model. Even though PyTorch docs sometimes still mention PYTORCH_CUDA_ALLOC_CONF, during runtime we get a deprecation warning.

[W1120 14:44:07.101109672 AllocatorConfig.cpp:28] Warning: PYTORCH_CUDA_ALLOC_CONF is deprecated, use PYTORCH_ALLOC_CONF instead

Solution

Use PYTORCH_ALLOC_CONF instead of PYTORCH_CUDA_ALLOC_CONF.

Result

Suppression of the deprecation warning.

@p-e-w p-e-w merged commit 2e1bb4b into p-e-w:master Nov 21, 2025
4 checks passed
@p-e-w
Copy link
Owner

p-e-w commented Nov 21, 2025

Thanks!

daggerstuff pushed a commit to daggerstuff/heretic that referenced this pull request Feb 22, 2026
…itecture-tRi3B

Add architecture improvement plan from code review
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