Enable CI builds with nvComp

Signed-off-by: Kamil Tokarski <ktokarski@nvidia.com>

Author: stiepan

conda/recipe/build.sh

@@ -83,6 +83,7 @@ cmake -DCUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
       -DBUILD_NVDEC=${BUILD_NVDEC:-ON}                    \
       -DBUILD_NVML=${BUILD_NVML:-ON}                      \
       -DBUILD_CUFILE=${BUILD_CUFILE:-ON}                  \
+      -DBUILD_NVCOMP=${BUILD_NVCOMP:-ON}                  \
       -DLINK_LIBCUDA=${LINK_LIBCUDA:-OFF}                 \
       -DWITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT:-${WITH_DYNAMIC_CUDA_TOOLKIT_DEFAULT}}\
       -DVERBOSE_LOGS=${VERBOSE_LOGS:-OFF}                 \

docker/Dockerfile

@@ -74,6 +74,8 @@ ARG BUILD_NVML
 ENV BUILD_NVML=${BUILD_NVML}
 ARG BUILD_CUFILE
 ENV BUILD_CUFILE=${BUILD_CUFILE}
+ARG BUILD_NVCOMP
+ENV BUILD_NVCOMP=${BUILD_NVCOMP}
 ARG LINK_DRIVER
 ENV LINK_DRIVER=${LINK_DRIVER}
 ARG WITH_DYNAMIC_CUDA_TOOLKIT

docker/Dockerfile.cuda102.x86_64.deps

@@ -12,5 +12,14 @@ RUN curl -LO https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_
     ./cuda_*.run --extract=/tmp/cuda/ --override && mv /tmp/cuda/cuda-toolkit/ /usr/local/cuda && \
     rm -f cuda_*.run;
 
+RUN NVCOMP_VERSION=2.4 && \
+    mkdir nvcomp && \
+    cd nvcomp && \
+    wget https://developer.download.nvidia.com/compute/nvcomp/${NVCOMP_VERSION}/local_installers/nvcomp_install_CUDA_10.2.tgz && \
+    cp -rv include/nvcomp* /usr/local/include/ && \
+    cp -v lib/* /usr/local/lib/ && \
+    cd .. && \
+    rm -rf nvcomp
+
 FROM scratch
 COPY --from=cuda /usr/local/cuda /usr/local/cuda

docker/Dockerfile.cuda118.aarch64.deps

@@ -10,3 +10,12 @@ RUN curl -LO https://developer.download.nvidia.com/compute/cuda/11.8.0/local_ins
     chmod +x cuda_*.run && \
     ./cuda_*.run --silent --no-opengl-libs --toolkit && \
     rm -f cuda_*.run;
+
+RUN NVCOMP_VERSION=2.4 && \
+    mkdir nvcomp && \
+    cd nvcomp && \
+    wget https://developer.download.nvidia.com/compute/nvcomp/${NVCOMP_VERSION}/local_installers/nvcomp_${NVCOMP_VERSION}_Linux_SBSA_CUDA_11.4.tgz && \
+    cp -rv include/nvcomp* /usr/local/include/ && \
+    cp -v lib/* /usr/local/lib/ && \
+    cd .. && \
+    rm -rf nvcomp

docker/Dockerfile.cuda118.x86_64.deps

@@ -11,6 +11,15 @@ RUN curl -LO https://developer.download.nvidia.com/compute/cuda/11.8.0/local_ins
     ./cuda_*.run --silent --no-opengl-libs --toolkit && \
     rm -f cuda_*.run;
 
+RUN NVCOMP_VERSION=2.4 && \
+    mkdir nvcomp && \
+    cd nvcomp && \
+    wget https://developer.download.nvidia.com/compute/nvcomp/${NVCOMP_VERSION}/local_installers/nvcomp_install_CUDA_11.x.tgz && \
+    cp -rv include/nvcomp* /usr/local/include/ && \
+    cp -v lib/* /usr/local/lib/ && \
+    cd .. && \
+    rm -rf nvcomp
+
 RUN NVJPEG2K_VERSION=0.6.0.28-1 && \
     CUFILE_VERSION=1.4.0.31-1 && \
     apt-get update && \

docker/build.sh

@@ -169,6 +169,7 @@ if [ "$BUILD_INHOST" == "YES" ]; then
                                         BUILD_NVML=${BUILD_NVML}                  \
                                         BUILD_FFTS=${BUILD_FFTS}                  \
                                         BUILD_CUFILE=${BUILD_CUFILE}              \
+                                        BUILD_NVCOMP=${BUILD_NVCOMP}              \
                                         LINK_DRIVER=${LINK_DRIVER}                \
                                         WITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT} \
                                         STRIP_BINARY=${STRIP_BINARY}              \
@@ -210,6 +211,7 @@ else
                                    --build-arg "BUILD_NVML=${BUILD_NVML}"                  \
                                    --build-arg "BUILD_FFTS=${BUILD_FFTS}"                  \
                                    --build-arg "BUILD_CUFILE=${BUILD_CUFILE}"              \
+                                   --build-arg "BUILD_NVCOMP=${BUILD_NVCOMP}"              \
                                    --build-arg "LINK_DRIVER=${LINK_DRIVER}"                \
                                    --build-arg "WITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT}"\
                                    --build_arg "STRIP_BINARY=${STRIP_BINARY}"              \

docker/build_helper.sh

@@ -44,6 +44,7 @@ export BUILD_LIBTAR=${BUILD_LIBTAR:-ON}
 export BUILD_NVML=${BUILD_NVML:-ON}
 export BUILD_FFTS=${BUILD_FFTS:-ON}
 export BUILD_CUFILE=${BUILD_CUFILE-OFF}
+export BUILD_NVCOMP=${BUILD_NVCOMP-OFF}
 export LINK_LIBCUDA=${LINK_LIBCUDA:-OFF}
 export LINK_CUDA_DYNAMICALLY=${LINK_CUDA_DYNAMICALLY:-OFF}
 export WITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT:-OFF}
@@ -89,6 +90,7 @@ cmake ../ -DCMAKE_INSTALL_PREFIX=.                 \
       -DBUILD_NVML=${BUILD_NVML}                   \
       -DBUILD_FFTS=${BUILD_FFTS}                   \
       -DBUILD_CUFILE=${BUILD_CUFILE}               \
+      -DBUILD_NVCOMP=${BUILD_NVCOMP}               \
       -DLINK_LIBCUDA=${LINK_LIBCUDA}               \
       -DLINK_CUDA_DYNAMICALLY=${LINK_CUDA_DYNAMICALLY} \
       -DWITH_DYNAMIC_CUDA_TOOLKIT=${WITH_DYNAMIC_CUDA_TOOLKIT} \