From c84e1e8b3dde5be9c3c095f5cf89a5c181848b5d Mon Sep 17 00:00:00 2001 From: Bradley Dice Date: Fri, 23 Feb 2024 05:03:07 -0600 Subject: [PATCH] Raise an error on import for unsupported GPUs. (#15053) RAPIDS 24.02 dropped support for Pascal GPUs. When using an unsupported GPU, the behavior of cudf is undefined and sometimes produces results that appear valid (and empty) but conceal CUDA kernel launch errors. This PR changes the behavior to error on import if unsupported GPUs are detected. Authors: - Bradley Dice (https://github.com/bdice) Approvers: - Matthew Roeschke (https://github.com/mroeschke) URL: https://github.com/rapidsai/cudf/pull/15053 --- python/cudf/cudf/utils/gpu_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/cudf/cudf/utils/gpu_utils.py b/python/cudf/cudf/utils/gpu_utils.py index 10a2f700cbd..b5387ddeb5f 100644 --- a/python/cudf/cudf/utils/gpu_utils.py +++ b/python/cudf/cudf/utils/gpu_utils.py @@ -86,7 +86,7 @@ def validate_setup(): minor_version = getDeviceAttribute( cudaDeviceAttr.cudaDevAttrComputeCapabilityMinor, 0 ) - warnings.warn( + raise UnsupportedCUDAError( "A GPU with NVIDIA Volta™ (Compute Capability 7.0) " "or newer architecture is required.\n" f"Detected GPU 0: {device_name}\n"