From 84ce6b552499c24239fb9d85c4d1217683094e77 Mon Sep 17 00:00:00 2001 From: pyynb Date: Thu, 1 Aug 2024 08:27:39 +0000 Subject: [PATCH 1/2] support torch2.4&cuda12.4 --- script/create_dev_conda_env.sh | 2 +- script/dgl_dev.yml.template | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/script/create_dev_conda_env.sh b/script/create_dev_conda_env.sh index 01a8795f63ec..891be0595400 100644 --- a/script/create_dev_conda_env.sh +++ b/script/create_dev_conda_env.sh @@ -1,6 +1,6 @@ #!/bin/bash -readonly CUDA_VERSIONS="11.8,12.1" +readonly CUDA_VERSIONS="11.8,12.1,12.4" readonly TORCH_VERSION="2.1.0" readonly PYTHON_VERSION="3.10" diff --git a/script/dgl_dev.yml.template b/script/dgl_dev.yml.template index b41d4a3db8c9..708df84aa1e5 100644 --- a/script/dgl_dev.yml.template +++ b/script/dgl_dev.yml.template @@ -10,7 +10,7 @@ dependencies: - pandoc - pygraphviz - pip: - - --find-links https://download.pytorch.org/whl/torch_stable.html + - --find-links https://download.pytorch.org/whl/torch/ - cmake>=3.18 - cython - filelock From e415839cc04d3a5d8cb5acd904a140cb0bc31639 Mon Sep 17 00:00:00 2001 From: pyynb Date: Fri, 2 Aug 2024 03:24:15 +0000 Subject: [PATCH 2/2] fix docstring typo --- python/dgl/graphbolt/impl/gpu_cached_feature.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/dgl/graphbolt/impl/gpu_cached_feature.py b/python/dgl/graphbolt/impl/gpu_cached_feature.py index 7a86965e6888..d8fb36add1da 100644 --- a/python/dgl/graphbolt/impl/gpu_cached_feature.py +++ b/python/dgl/graphbolt/impl/gpu_cached_feature.py @@ -89,7 +89,7 @@ def read(self, ids: torch.Tensor = None): return values def read_async(self, ids: torch.Tensor): - """Read the feature by index asynchronously. + r"""Read the feature by index asynchronously. Parameters ---------- @@ -104,7 +104,7 @@ def read_async(self, ids: torch.Tensor): can be accessed by calling `.wait()`. on the returned future object. It is undefined behavior to call `.wait()` more than once. - Example Usage + Examples -------- >>> import dgl.graphbolt as gb >>> feature = gb.Feature(...)