Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[BUG] Skip WholeGraph Tests if GPU PyTorch Unavailable #4820

Merged
merged 16 commits into from
Jan 3, 2025
Merged

[BUG] Skip WholeGraph Tests if GPU PyTorch Unavailable #4820

Changes from all commits
Commits
Show all changes
16 commits
Select commit Hold shift + click to select a range
7e4a15e
set max pt version, skip 11.4 tests using wg
alexbarghi-nv Dec 5, 2024
9ad4278
Add alpha limit
alexbarghi-nv Dec 6, 2024
5574c41
use different function for getting cudart version
alexbarghi-nv Dec 6, 2024
03d63ae
Merge branch 'skip-114' of https://github.com/alexbarghi-nv/cugraph i…
alexbarghi-nv Dec 6, 2024
85fa7fe
rerun generator
alexbarghi-nv Dec 6, 2024
5c58965
restore wholegraph check
alexbarghi-nv Dec 6, 2024
0635b77
Merge branch 'branch-24.12' of https://github.com/rapidsai/cugraph in…
alexbarghi-nv Dec 9, 2024
a9d0c2d
x
alexbarghi-nv Dec 9, 2024
cdccd78
simplify torch check, don't check if missing
alexbarghi-nv Dec 10, 2024
4bc3bb9
[BUG] Skip WholeGraph Tests if GPU PyTorch Unavailable
jameslamb Dec 12, 2024
5ea750c
Merge branch 'branch-25.02' into skip-cpu
jameslamb Dec 17, 2024
1fe9ecb
Merge branch 'branch-25.02' into skip-cpu
alexbarghi-nv Jan 2, 2025
e03e958
Merge branch 'skip-cpu' of https://github.com/alexbarghi-nv/cugraph i…
alexbarghi-nv Jan 2, 2025
d9c71e8
add pytestmark
alexbarghi-nv Jan 2, 2025
ef10c4b
change copyright
alexbarghi-nv Jan 3, 2025
e51be9c
change copyright
alexbarghi-nv Jan 3, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
30 changes: 15 additions & 15 deletions python/cugraph/cugraph/tests/data_store/test_gnn_feat_storage_wholegraph.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright (c) 2023-2024, NVIDIA CORPORATION.
# Copyright (c) 2023-2025, NVIDIA CORPORATION.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
Expand Down Expand Up @@ -32,6 +32,20 @@ def get_cudart_version():
return major * 1000 + minor * 10


pytestmark = [
pytest.mark.skipif(
isinstance(torch, MissingModule) or not torch.cuda.is_available(),
reason="PyTorch with GPU support not available",
),
pytest.mark.skipif(
isinstance(pylibwholegraph, MissingModule), reason="wholegraph not available"
),
pytest.mark.skipif(
get_cudart_version() < 11080, reason="not compatible with CUDA < 11.8"
),
]


def runtest(rank: int, world_size: int):
torch.cuda.set_device(rank)

Expand Down Expand Up @@ -69,13 +83,6 @@ def runtest(rank: int, world_size: int):


@pytest.mark.sg
@pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
@pytest.mark.skipif(
isinstance(pylibwholegraph, MissingModule), reason="wholegraph not available"
)
@pytest.mark.skipif(
get_cudart_version() < 11080, reason="not compatible with CUDA < 11.8"
)
def test_feature_storage_wholegraph_backend():
world_size = torch.cuda.device_count()
print("gpu count:", world_size)
Expand All @@ -87,13 +94,6 @@ def test_feature_storage_wholegraph_backend():


@pytest.mark.mg
@pytest.mark.skipif(isinstance(torch, MissingModule), reason="torch not available")
@pytest.mark.skipif(
isinstance(pylibwholegraph, MissingModule), reason="wholegraph not available"
)
@pytest.mark.skipif(
get_cudart_version() < 11080, reason="not compatible with CUDA < 11.8"
)
def test_feature_storage_wholegraph_backend_mg():
world_size = torch.cuda.device_count()
print("gpu count:", world_size)
Expand Down
Loading