CI: skip tests marked "cufile" on cuda 11.8 and arm64

rapidsai · Oct 25, 2024 · 2ad401f · 2ad401f
1 parent 4b8181c
commit 2ad401f
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/ci/run_pytests.sh b/ci/run_pytests.sh
@@ -6,4 +6,11 @@ set -euo 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".
+# cuFile didn't support arm until 12.4
+PYTEST_MARK=$(
+  [[ "${CUDA_VERSION}" == "11.8.0" && "${NVARCH}" == "sbsa" ]] \
+  && echo "-m 'not cufile'" || echo " "
+)
+
+python -m pytest ${PYTEST_MARK} --cache-clear --verbose "$@" tests
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
@@ -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 ./python/kvikio/tests
+# If running CUDA 11.8 on arm64, we skip tests marked "cufile".
+# cuFile didn't support arm until 12.4
+PYTEST_MARK=$(
+  [[ "${CUDA_VERSION}" == "11.8.0" && "${NVARCH}" == "sbsa" ]] \
+  && echo "-m 'not cufile'" || echo " "
+)
+
+python -m pytest ${PYTEST_MARK} ./python/kvikio/tests
diff --git a/cpp/include/kvikio/shim/libcurl.hpp b/cpp/include/kvikio/shim/libcurl.hpp
@@ -224,6 +224,7 @@ class CurlHandle {
       } else {
         ss << "(" << msg << ")";
       }
+      std::cout << "perform() error: " << ss.str() << " (ERROR END)" << std::endl;
       throw std::runtime_error(ss.str());
     }
   }