Share GPU diagnostics across CI tests

Author: bdice

ci/gpu_info.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+set -euo pipefail
+
+rapids-logger "Check GPU status"
+nvidia-smi
+
+rapids-logger "Check memory configuration"
+nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
+grep Coherent /proc/driver/nvidia/params || echo "Coherent GPU memory mode not reported"
+grep -o 'init_on_alloc=[^ ]*' /proc/cmdline || echo "init_on_alloc: kernel default"

ci/test_cpp.sh

@@ -33,13 +33,7 @@ set -u
 
 rapids-print-env
 
-rapids-logger "Check GPU status"
-nvidia-smi
-
-rapids-logger "Check memory configuration"
-nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
-grep Coherent /proc/driver/nvidia/params || echo "Coherent GPU memory mode not reported"
-grep -o 'init_on_alloc=[^ ]*' /proc/cmdline || echo "init_on_alloc: kernel default"
+./ci/gpu_info.sh
 
 # Run librmm gtests from librmm-tests package
 rapids-logger "Run gtests"

ci/test_cpp_debug.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
@@ -33,9 +33,7 @@ set -u
 
 rapids-print-env
 
-rapids-logger "Check GPU status"
-nvidia-smi
-nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
+./ci/gpu_info.sh
 
 rapids-logger "Building librmm in Debug mode"
 cmake -S cpp -B cpp/build \

ci/test_python.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2020-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
@@ -39,9 +39,7 @@ RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}
 RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${PWD}/coverage-results"}
 mkdir -p "${RAPIDS_TESTS_DIR}" "${RAPIDS_COVERAGE_DIR}"
 
-rapids-logger "Check GPU status"
-nvidia-smi
-nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
+./ci/gpu_info.sh
 
 rapids-logger "pytest rmm"
 

ci/test_python_integrations.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -euo pipefail
@@ -22,9 +22,7 @@ RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${PWD}/test-results"}
 RAPIDS_COVERAGE_DIR=${RAPIDS_COVERAGE_DIR:-"${PWD}/coverage-results"}
 mkdir -p "${RAPIDS_TESTS_DIR}" "${RAPIDS_COVERAGE_DIR}"
 
-rapids-logger "Check GPU status"
-nvidia-smi
-nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
+./ci/gpu_info.sh
 
 EXITCODE=0
 

ci/test_wheel_integrations.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION.
+# SPDX-FileCopyrightText: Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
 
 set -eou pipefail
@@ -28,9 +28,7 @@ PIP_INSTALL_SHARED_ARGS=(
 
 EXITCODE=0
 
-rapids-logger "Check GPU status"
-nvidia-smi
-nvidia-smi -q | grep "Addressing Mode" || echo "Addressing Mode not reported"
+./ci/gpu_info.sh
 
 echo "::group::PyTorch Tests"