feat: Nvidia GPU Direct Storage Support for reading RNTuple

pyproject.toml

@@ -91,6 +91,20 @@ readme = "README.md"
 requires-python = ">=3.9"
 
 [project.optional-dependencies]
+GDS_cu11 = [
+  "kvikio-cu11>=25.02.01"
+]
+GDS_cu12 = [
+  "kvikio-cu12>=25.02.01"
+]
+dev = [
+  "boost_histogram>=0.13",
+  "dask-awkward>=2025.2.0",
+  "dask[array,distributed]",
+  "hist>=1.2",
+  "pandas",
+  "awkward-pandas"
+]
 http = ["aiohttp"]
 s3 = ["s3fs"]
 xrootd = ["fsspec-xrootd>=0.5.0"]

src/uproot/extras.py

@@ -345,3 +345,51 @@ def awkward_pandas():
         ) from err
     else:
         return awkward_pandas
+
+
+def cupy():
+    """
+    Imports and returns ``cupy``.
+    """
+    try:
+        import cupy
+    except ModuleNotFoundError as err:
+        raise ModuleNotFoundError(
+            """Cupy is required for GDS reading to work. Please install GDS dependencies with:
+    `python3 -m pip install uproot[GDS_cuX]`
+where X is the cuda major version on user's system (11 and 12 currently supported). Cuda major version can be checked by calling `nvidia-smi --version` or `nvcc --version` if available."""
+        ) from err
+    else:
+        return cupy
+
+
+def kvikio():
+    """
+    Imports and returns ``kvikio``.
+    """
+    try:
+        import kvikio
+    except ModuleNotFoundError as err:
+        raise ModuleNotFoundError(
+            """Kvikio is required for GDS reading to work. Please install GDS dependencies with:
+    `python3 -m pip install uproot[GDS_cuX]`
+where X is the cuda major version on user's system (11 and 12 currently supported). Cuda major version can be checked by calling `nvidia-smi --version` or `nvcc --version` if available."""
+        ) from err
+    else:
+        return kvikio
+
+
+def kvikio_nvcomp_codec():
+    """
+    Imports and returns ``kvikio``.
+    """
+    try:
+        import kvikio.nvcomp_codec as kvikio_nvcomp_codec
+    except ModuleNotFoundError as err:
+        raise ModuleNotFoundError(
+            """Kvikio is required for GDS reading to work. Please install GDS dependencies with:
+    `python3 -m pip install uproot[GDS_cuX]`
+where X is the cuda major version on user's system (11 and 12 currently supported). Cuda major version can be checked by calling `nvidia-smi --version` or `nvcc --version` if available."""
+        ) from err
+    else:
+        return kvikio_nvcomp_codec