Skip to content

Commit

Permalink
Make sure that GDS on SBSA is not tested for drivers below cuda 12.2
Browse files Browse the repository at this point in the history
- as the SBSA started supporting GDS from CUDA 12.2 we should not test
  this using driver which don't support it

Signed-off-by: Janusz Lisiecki <jlisiecki@nvidia.com>
  • Loading branch information
JanuszL committed Nov 20, 2023
1 parent 1cb4835 commit 14cdc62
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions dali/test/python/reader/test_numpy.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,18 +46,22 @@ def is_gds_supported(device_id=0):
return is_gds_supported_var

compute_cap = 0
cuda_drv_ver = 0
try:
import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(device_id)
compute_cap = pynvml.nvmlDeviceGetCudaComputeCapability(handle)
compute_cap = compute_cap[0] + compute_cap[1] / 10.
cuda_drv_ver = pynvml.nvmlSystemGetCudaDriverVersion()
except ModuleNotFoundError:
print("Python bindings for NVML not found")

# for CUDA < 12.2 only x86 platform is supported, above aarch64 is supported as well
is_gds_supported_var = (platform.processor() == "x86_64" or
dali_b.__cuda_version__ >= 12200) and compute_cap >= 6.0
is_gds_supported_var = (
platform.processor() == "x86_64"
or (dali_b.__cuda_version__ >= 12200 and cuda_drv_ver >= 12020)
) and compute_cap >= 6.0
return is_gds_supported_var


Expand Down

0 comments on commit 14cdc62

Please sign in to comment.