From d3d864fad29bb87f12e0275899e88a450edc8539 Mon Sep 17 00:00:00 2001
From: Georgii Evtushenko <evtushenko.georgy@gmail.com>
Date: Thu, 29 Feb 2024 15:50:38 -0800
Subject: [PATCH] Fix CUB docs image fetcher (#1466)

* Fix CUB docs image fetcher

* Add git to doc container
---
 cub/docs/gen_docs.bash | 6 +++++-
 docs/Dockerfile        | 2 +-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/cub/docs/gen_docs.bash b/cub/docs/gen_docs.bash
index 34ba0084252..d35ce5e2d7b 100755
--- a/cub/docs/gen_docs.bash
+++ b/cub/docs/gen_docs.bash
@@ -13,6 +13,10 @@ cd $SCRIPT_PATH
 rm -rf img
 mkdir -p img
 
+if [ ! -d cubimg ]; then
+    git clone -b gh-pages https://github.com/NVlabs/cub.git cubimg
+fi 
+
 if [ ! -n "$(find img -name '*.png')" ]; then
     wget -q https://docs.nvidia.com/cuda/_static/Logo_and_CUDA.png -O img/logo.png
 
@@ -23,7 +27,7 @@ if [ ! -n "$(find img -name '*.png')" ]; then
     for img in "${imgs[@]}"
     do
         echo ${img}
-        wget -q https://nvlabs.github.io/cub/${img} -O img/${img} || echo "!!! Failed to fetch $img"
+        cp cubimg/${img} img/${img}
     done
 fi
 
diff --git a/docs/Dockerfile b/docs/Dockerfile
index f628b1b7f8f..2b01d3c8eb2 100644
--- a/docs/Dockerfile
+++ b/docs/Dockerfile
@@ -6,7 +6,7 @@ ENV APT_KEY_DONT_WARN_ON_DANGEROUS_USAGE=1
 
 RUN apt-get -y -qq update; \
     apt-get -y -qq upgrade; \
-    apt-get -y -qq install doxygen unzip wget
+    apt-get -y -qq install doxygen unzip wget git
 
 RUN mkdir -p /opt/doxybook2; \
     cd /opt/doxybook2; \