pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[GraphBolt][CUDA] Fix error when empty env variable is used. (#7748)

Browse Source
This commit is contained in:
Muhammed Fatih BALIN
2024-08-26 20:28:48 -04:00
committed by GitHub
parent 6bce0cdfc5
commit f38790bfba
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/dgl/graphbolt/__init__.py
View File

@@ -15,7 +15,7 @@ cuda_allocator_env = os.getenv("PYTORCH_CUDA_ALLOC_CONF")
WARNING_STR_TO_BE_SHOWN = None
configs = (
{}
if cuda_allocator_env is None
if cuda_allocator_env is None or len(cuda_allocator_env) == 0
else {
kv_pair.split(":")[0]: kv_pair.split(":")[1]
for kv_pair in cuda_allocator_env.split(",")