From a2d534f4c3f9ef8b4b27cfa08ae9b288f7004b6b Mon Sep 17 00:00:00 2001
From: Georgii Evtushenko <evtushenko.georgy@gmail.com>
Date: Thu, 29 Feb 2024 17:35:51 +0000
Subject: [PATCH] Fix CUB docs image fetcher

---
 cub/docs/gen_docs.bash | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/cub/docs/gen_docs.bash b/cub/docs/gen_docs.bash
index 34ba008425..d35ce5e2d7 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