File tree Expand file tree Collapse file tree 2 files changed +18
-3
lines changed
Expand file tree Collapse file tree 2 files changed +18
-3
lines changed Original file line number Diff line number Diff line change @@ -87,9 +87,7 @@ def ipc_device():
8787 pytest .skip ("Device does not support IPC" )
8888
8989 # Skip on WSL or if driver rejects IPC-enabled mempool creation on this platform/device
90- from cuda_python_test_helpers import IS_WSL , supports_ipc_mempool
91-
92- if IS_WSL or not supports_ipc_mempool (device ):
90+ if helpers .IS_WSL or not helpers .supports_ipc_mempool (device ):
9391 pytest .skip ("Driver rejects IPC-enabled mempool creation on this platform" )
9492
9593 return device
Original file line number Diff line number Diff line change 22# SPDX-License-Identifier: Apache-2.0
33
44import os
5+ import pathlib
6+ import sys
57
68CUDA_PATH = os .environ .get ("CUDA_PATH" )
79CUDA_INCLUDE_PATH = None
1416 path = os .path .join (path , "cccl" )
1517 if os .path .isdir (path ):
1618 CCCL_INCLUDE_PATHS = (path ,) + CCCL_INCLUDE_PATHS
19+
20+
21+ try :
22+ import cuda_python_test_helpers
23+ except ImportError :
24+ # Import shared platform helpers for tests across repos
25+ sys .path .insert (0 , str (pathlib .Path (__file__ ).resolve ().parents [2 ] / "cuda_python_test_helpers" ))
26+ import cuda_python_test_helpers
27+
28+
29+ IS_WSL = cuda_python_test_helpers .IS_WSL
30+ supports_ipc_mempool = cuda_python_test_helpers .supports_ipc_mempool
31+
32+
33+ del cuda_python_test_helpers
You can’t perform that action at this time.
0 commit comments