Skip to content

Commit ae70831

Browse files
committed
Switch to newer API
1 parent c2e6560 commit ae70831

1 file changed

Lines changed: 6 additions & 9 deletions

File tree

qa/common/test_util.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
#!/usr/bin/env python3
22

3-
# Copyright 2018-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3+
# Copyright 2018-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
44
#
55
# Redistribution and use in source and binary forms, with or without
66
# modification, are permitted provided that the following conditions
@@ -313,16 +313,13 @@ def check_gpus_compute_capability(min_capability):
313313
import importlib.util
314314

315315
if importlib.util.find_spec("cuda") is not None:
316-
import cuda.core.experimental as cuda_core_experimental
317-
318-
devices = cuda_core_experimental.system.devices
316+
from cuda.core import Device
319317

318+
devices = Device.get_all_devices()
320319
for device in devices:
321-
compute_capability = (
322-
device.compute_capability.major + device.compute_capability.minor / 10.0
323-
)
324-
325-
if compute_capability < min_capability:
320+
cc = device.compute_capability
321+
compute_capability_value = cc.major + cc.minor / 10.0
322+
if compute_capability_value < min_capability:
326323
return False
327324

328325
elif importlib.util.find_spec("pycuda") is not None:

0 commit comments

Comments
 (0)