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

Correcting the misleading reason for skipping the test. (#7749)

Browse Source
This commit is contained in:
Andrei Ivanov
2024-08-30 10:18:16 -07:00
committed by GitHub
parent a71946539f
commit bb02829b6a
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/python/pytorch/graphbolt/impl/test_gpu_graph_cache.py
View File

@@ -11,7 +11,9 @@ import torch
@unittest.skipIf(
F._default_context_str != "gpu"
or torch.cuda.get_device_capability()[0] < 7,
reason="GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
reason="GPUCachedFeature tests are available only on GPU."
if F._default_context_str != "gpu"
else "GPUCachedFeature requires a Volta or later generation NVIDIA GPU.",
)
@pytest.mark.parametrize(
"indptr_dtype",