File tree Expand file tree Collapse file tree
cuda/bindings/_test_helpers Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -42,6 +42,8 @@ def hardware_supports_nvml():
4242 finally :
4343 if initialized :
4444 nvml .shutdown ()
45+
46+
4547@contextmanager
4648def unsupported_before (device : int , expected_device_arch : nvml .DeviceArch | str | None ):
4749 device_arch = nvml .device_get_architecture (device )
Original file line number Diff line number Diff line change 55import sys
66from importlib .metadata import PackageNotFoundError , distribution
77
8- import pytest
9-
108import cuda .bindings .driver as cuda
9+ import pytest
1110from cuda .bindings ._test_helpers .arch_check import hardware_supports_nvml
1211
1312# Import shared test helpers for tests across subprojects.
Original file line number Diff line number Diff line change 33
44
55import pytest
6-
76from cuda .bindings import nvml
87from cuda .bindings ._test_helpers import arch_check
98from cuda .pathfinder import DynamicLibNotFoundError
You can’t perform that action at this time.
0 commit comments