Skip to content

Commit 5def54c

Browse files
committed
remove cuda env
Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent 5c568ec commit 5def54c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm_ascend/device_allocator/camem.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def get_instance() -> "CaMemAllocator":
144144
return CaMemAllocator.instance
145145

146146
def __init__(self):
147-
conf = os.environ.get("PYTORCH_CUDA_ALLOC_CONF", "")
147+
conf = os.environ.get("PYTORCH_NPU_ALLOC_CONF", "")
148148
assert "expandable_segments:True" not in conf, \
149149
("Expandable segments are not compatible with memory pool. "
150150
"Please track https://github.com/pytorch/pytorch/issues/147851 "

0 commit comments

Comments
 (0)