We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5c568ec commit 5def54cCopy full SHA for 5def54c
vllm_ascend/device_allocator/camem.py
@@ -144,7 +144,7 @@ def get_instance() -> "CaMemAllocator":
144
return CaMemAllocator.instance
145
146
def __init__(self):
147
- conf = os.environ.get("PYTORCH_CUDA_ALLOC_CONF", "")
+ conf = os.environ.get("PYTORCH_NPU_ALLOC_CONF", "")
148
assert "expandable_segments:True" not in conf, \
149
("Expandable segments are not compatible with memory pool. "
150
"Please track https://github.com/pytorch/pytorch/issues/147851 "
0 commit comments