From c07e16295a47426c1da50e2191b62f4cad1d6ef8 Mon Sep 17 00:00:00 2001
From: Ray Douglass <3107146+raydouglass@users.noreply.github.com>
Date: Fri, 1 Apr 2022 15:03:01 -0400
Subject: [PATCH] Only run doc builds for x86_64 (#4680)

We don't have arm64 packages for `rapids-doc-env`, so only run the doc builds for x86_64

Authors:
   - Ray Douglass (https://github.com/raydouglass)

Approvers:
   - AJ Schmidt (https://github.com/ajschmidt8)
---
 ci/gpu/build.sh | 22 ++++++++++++++--------
 1 file changed, 14 insertions(+), 8 deletions(-)

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index f89e2c80b2..2f254836ec 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -69,9 +69,13 @@ gpuci_mamba_retry install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvid
       "xgboost=1.5.2dev.rapidsai${MINOR_VERSION}" \
       "rapids-build-env=${MINOR_VERSION}.*" \
       "rapids-notebook-env=${MINOR_VERSION}.*" \
-      "rapids-doc-env=${MINOR_VERSION}.*" \
       "shap>=0.37,<=0.39"
 
+if [ "$(arch)" = "x86_64" ]; then
+    gpuci_mamba_retry install -c conda-forge -c rapidsai -c rapidsai-nightly -c nvidia \
+        "rapids-doc-env=${MINOR_VERSION}.*"
+fi
+
 # https://docs.rapids.ai/maintainers/depmgmt/
 # gpuci_conda_retry remove --force rapids-build-env rapids-notebook-env
 # gpuci_mamba_retry install -y "your-pkg=1.0.0"
@@ -279,14 +283,16 @@ else
         python ../scripts/cuda-memcheck.py -tool memcheck -exe ./test/prims
     fi
 
-    gpuci_logger "Building doxygen C++ docs"
-    #Need to run in standard directory, not our artifact dir
-    unset LIBCUML_BUILD_DIR
-    $WORKSPACE/build.sh cppdocs -v
+    if [ "$(arch)" = "x86_64" ]; then
+        gpuci_logger "Building doxygen C++ docs"
+        #Need to run in standard directory, not our artifact dir
+        unset LIBCUML_BUILD_DIR
+        $WORKSPACE/build.sh cppdocs -v
 
-    if [ "$CUDA_REL" != "11.0" ]; then
-        gpuci_logger "Building python docs"
-        $WORKSPACE/build.sh pydocs
+        if [ "$CUDA_REL" != "11.0" ]; then
+            gpuci_logger "Building python docs"
+            $WORKSPACE/build.sh pydocs
+        fi
     fi
 
 fi