From 3f8f8b7929850f2135c5d5dbe48f24f17c0a6501 Mon Sep 17 00:00:00 2001
From: "Mads R. B. Kristensen" <madsbk@gmail.com>
Date: Sun, 27 Oct 2024 11:01:30 +0100
Subject: [PATCH] ci debug

---
 ci/run_pytests.sh | 11 +++++++++--
 ci/test_wheel.sh  | 11 +++++++++--
 2 files changed, 18 insertions(+), 4 deletions(-)

diff --git a/ci/run_pytests.sh b/ci/run_pytests.sh
index b2c93dbe56..2732accf97 100755
--- a/ci/run_pytests.sh
+++ b/ci/run_pytests.sh
@@ -1,9 +1,16 @@
 #!/bin/bash
 # Copyright (c) 2024, NVIDIA CORPORATION.
 
-set -euo pipefail
+set -xeou pipefail
 
 # Support invoking run_pytests.sh outside the script directory
 cd "$(dirname "$(realpath "${BASH_SOURCE[0]}")")"/../python/kvikio
 
-pytest --cache-clear --verbose "$@" tests
+# If running CUDA 11.8 on arm64, we skip tests marked "cufile" since
+# cuFile didn't support arm until 12.4
+[[ "${CUDA_VERSION}" == "11.8.0" && "${RUNNER_ARCH}" == "ARM64" ]] \
+  && PYTEST_MARK=( -m 'not cufile' ) || PYTEST_MARK=()
+
+export
+
+pytest --cache-clear --verbose "${PYTEST_MARK[@]}" "$@" tests
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 301c27188a..2a29bf84dd 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -1,7 +1,7 @@
 #!/bin/bash
 # Copyright (c) 2023-2024, NVIDIA CORPORATION.
 
-set -eou pipefail
+set -xeou pipefail
 
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 WHEELHOUSE="${PWD}/dist/"
@@ -9,4 +9,11 @@ RAPIDS_PY_WHEEL_NAME="kvikio_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-fr
 
 python -m pip install "$(echo ${WHEELHOUSE}/kvikio_${RAPIDS_PY_CUDA_SUFFIX}*.whl)[test]"
 
-python -m pytest --verbose ./python/kvikio/tests
+# If running CUDA 11.8 on arm64, we skip tests marked "cufile" since
+# cuFile didn't support arm until 12.4
+[[ "${CUDA_VERSION}" == "11.8.0" && "${RUNNER_ARCH}" == "ARM64" ]] \
+  && PYTEST_MARK=( -m 'not cufile' ) || PYTEST_MARK=()
+
+export
+
+python -m pytest --cache-clear --verbose "${PYTEST_MARK[@]}" ./python/kvikio/tests