Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
dgl
Commits
d853a010
Unverified
Commit
d853a010
authored
Apr 18, 2024
by
Muhammed Fatih BALIN
Committed by
GitHub
Apr 19, 2024
Browse files
[GraphBolt] Enable `DistributedItemSampler` test only on CPU. (#7323)
parent
9a340c77
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
1 deletion
+4
-1
tests/python/pytorch/graphbolt/test_item_sampler.py
tests/python/pytorch/graphbolt/test_item_sampler.py
+4
-1
No files found.
tests/python/pytorch/graphbolt/test_item_sampler.py
View file @
d853a010
import
os
import
re
import
unittest
from
sys
import
platform
import
backend
as
F
import
dgl
import
pytest
import
torch
import
torch.distributed
as
dist
import
torch.multiprocessing
as
mp
from
dgl
import
graphbolt
as
gb
from
torch.testing
import
assert_close
def
test_ItemSampler_minibatcher
():
...
...
@@ -1090,6 +1092,7 @@ def test_RangeCalculation(params):
assert
key
==
answer
@
unittest
.
skipIf
(
F
.
_default_context_str
!=
"cpu"
,
reason
=
"GPU not required."
)
@
pytest
.
mark
.
parametrize
(
"num_ids"
,
[
24
,
30
,
32
,
34
,
36
])
@
pytest
.
mark
.
parametrize
(
"num_workers"
,
[
0
,
2
])
@
pytest
.
mark
.
parametrize
(
"drop_last"
,
[
False
,
True
])
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment