Mark ucx-py tests for GPU (dask#5133)

mrocklin · Jul 29, 2021 · 1be9265 · 1be9265
1 parent a5ccc8f
commit 1be9265
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/distributed/comm/tests/test_ucx.py b/distributed/comm/tests/test_ucx.py
@@ -2,6 +2,8 @@
 
 import pytest
 
+pytestmark = pytest.mark.gpu
+
 ucp = pytest.importorskip("ucp")
 
 from distributed import Client, Scheduler, wait

diff --git a/distributed/comm/tests/test_ucx_config.py b/distributed/comm/tests/test_ucx_config.py
@@ -2,6 +2,8 @@
 
 import pytest
 
+pytestmark = pytest.mark.gpu
+
 import dask
 from dask.utils import format_bytes