feat: Nvidia GPU Direct Storage Support for reading RNTuple

pyproject.toml

@@ -29,6 +29,7 @@ test = [
   "rangehttpserver",
   "requests",
   "s3fs",
+  'kvikio-cu12>=25.02.01; platform_system == "Linux" and python_version >= "3.10"',
   {include-group = "test-core"}
 ]
 test-core = [
@@ -76,7 +77,7 @@ dependencies = [
   "awkward>=2.4.6",
   "cramjam>=2.5.0",
   "xxhash",
-  "numpy",
+  "numpy < 2.3",
   "fsspec",
   "packaging",
   "typing_extensions>=4.1.0; python_version < '3.11'"
@@ -91,6 +92,12 @@ readme = "README.md"
 requires-python = ">=3.9"
 
 [project.optional-dependencies]
+GDS_cu11 = [
+  "kvikio-cu11>=25.02.01"
+]
+GDS_cu12 = [
+  "kvikio-cu12>=25.02.01"
+]
 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