Disable example and packages builds by default

NVIDIA · Sep 13, 2023 · 54fff7e · 54fff7e
1 parent f7e3663
commit 54fff7e
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 3 deletions.
diff --git a/docs/building_blocks.md b/docs/building_blocks.md
@@ -3252,6 +3252,12 @@ version 3.19 or later is required and must be installed separately.
 __Parameters__
 
 
+- __build_examples__: Boolean flag to specify whether the NVSHMEM
+examples should be built.  The default is False.
+
+- __build_packages__: Boolean flag to specify whether the RPM and deb
+packages should be built.  The default is False.
+
 - __cmake_opts__: List of additional options to pass to `cmake`.  The
 default value is an empty list.
 

diff --git a/hpccm/building_blocks/nvshmem.py b/hpccm/building_blocks/nvshmem.py
@@ -47,6 +47,12 @@ class nvshmem(bb_base, hpccm.templates.downloader, hpccm.templates.envvars,
 
     # Parameters
 
+    build_examples: Boolean flag to specify whether the NVSHMEM
+    examples should be built.  The default is False.
+
+    build_packages: Boolean flag to specify whether the RPM and deb
+    packages should be built.  The default is False.
+
     cmake_opts: List of additional options to pass to `cmake`.  The
     default value is an empty list.
 
@@ -93,6 +99,8 @@ def __init__(self, **kwargs):
 
         super(nvshmem, self).__init__(**kwargs)
 
+        self.__build_examples = kwargs.pop('build_examples', False)
+        self.__build_packages = kwargs.pop('build_packages', False)
         self.__cmake_opts = kwargs.pop('cmake_opts', [])
         self.__cuda = kwargs.pop('cuda', '/usr/local/cuda')
         self.__gdrcopy = kwargs.pop('gdrcopy', None)
@@ -138,6 +146,14 @@ def __init__(self, **kwargs):
     def __configure(self):
         """Setup build options based on user parameters"""
 
+        if self.__build_examples is False:
+            self.__cmake_opts.append('-DNVSHMEM_BUILD_EXAMPLES=OFF')
+
+        if self.__build_packages is False:
+            self.__cmake_opts.append('-DNVSHMEM_BUILD_PACKAGES=OFF')
+            self.__cmake_opts.append('-DNVSHMEM_BUILD_DEB_PACKAGES=OFF')
+            self.__cmake_opts.append('-DNVSHMEM_BUILD_RPM_PACKAGES=OFF')
+
         if self.__cuda:
             self.__cmake_opts.append('-DCUDA_HOME={}'.format(self.__cuda))
 

diff --git a/test/test_nvshmem.py b/test/test_nvshmem.py
@@ -45,7 +45,7 @@ def test_defaults_ubuntu(self):
     rm -rf /var/lib/apt/lists/*
 RUN mkdir -p /var/tmp && wget -q -nc --no-check-certificate -P /var/tmp https://developer.download.nvidia.com/compute/redist/nvshmem/2.9.0/source/nvshmem_src_2.9.0-2.txz && \
     mkdir -p /var/tmp && tar -x -f /var/tmp/nvshmem_src_2.9.0-2.txz -C /var/tmp -J && \
-    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DCUDA_HOME=/usr/local/cuda /var/tmp/nvshmem_src_2.9.0-2 && \
+    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DNVSHMEM_BUILD_EXAMPLES=OFF -DNVSHMEM_BUILD_PACKAGES=OFF -DNVSHMEM_BUILD_DEB_PACKAGES=OFF -DNVSHMEM_BUILD_RPM_PACKAGES=OFF -DCUDA_HOME=/usr/local/cuda /var/tmp/nvshmem_src_2.9.0-2 && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target all -- -j$(nproc) && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target install -- -j$(nproc) && \
     rm -rf /var/tmp/nvshmem_src_2.9.0-2 /var/tmp/nvshmem_src_2.9.0-2.txz
@@ -68,7 +68,7 @@ def test_package_ubuntu(self):
     rm -rf /var/lib/apt/lists/*
 COPY nvshmem_src_2.9.0-2.tar.xz /var/tmp/nvshmem_src_2.9.0-2.tar.xz
 RUN mkdir -p /var/tmp && tar -x -f /var/tmp/nvshmem_src_2.9.0-2.tar.xz -C /var/tmp -J && \
-    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DCUDA_HOME=/usr/local/cuda /var/tmp/nvshmem_src_2.9.0-2 && \
+    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DNVSHMEM_BUILD_EXAMPLES=OFF -DNVSHMEM_BUILD_PACKAGES=OFF -DNVSHMEM_BUILD_DEB_PACKAGES=OFF -DNVSHMEM_BUILD_RPM_PACKAGES=OFF -DCUDA_HOME=/usr/local/cuda /var/tmp/nvshmem_src_2.9.0-2 && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target all -- -j$(nproc) && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target install -- -j$(nproc) && \
     rm -rf /var/tmp/nvshmem_src_2.9.0-2 /var/tmp/nvshmem_src_2.9.0-2.tar.xz
@@ -95,7 +95,7 @@ def test_cmake_options_centos(self):
     rm -rf /var/cache/yum/*
 RUN mkdir -p /var/tmp && wget -q -nc --no-check-certificate -P /var/tmp https://developer.download.nvidia.com/compute/redist/nvshmem/2.9.0/source/nvshmem_src_2.9.0-2.txz && \
     mkdir -p /var/tmp && tar -x -f /var/tmp/nvshmem_src_2.9.0-2.txz -C /var/tmp -J && \
-    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DNVSHMEM_USE_NCCL=1 -DNVSHMEM_UCX_SUPPORT=1 -DCUDA_HOME=/usr/local/cuda -DGDRCOPY_HOME=/usr/local/gdrcopy -DNVSHMEM_MPI_SUPPORT= 1 -DMPI_HOME=/usr/local/openmpi -DNVSHMEM_SHMEM_SUPPORT=1 -DSHMEM_HOME=/usr/local/openmpi /var/tmp/nvshmem_src_2.9.0-2 && \
+    mkdir -p /var/tmp/nvshmem_src_2.9.0-2/build && cd /var/tmp/nvshmem_src_2.9.0-2/build && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/nvshmem -DNVSHMEM_USE_NCCL=1 -DNVSHMEM_UCX_SUPPORT=1 -DNVSHMEM_BUILD_EXAMPLES=OFF -DNVSHMEM_BUILD_PACKAGES=OFF -DNVSHMEM_BUILD_DEB_PACKAGES=OFF -DNVSHMEM_BUILD_RPM_PACKAGES=OFF -DCUDA_HOME=/usr/local/cuda -DGDRCOPY_HOME=/usr/local/gdrcopy -DNVSHMEM_MPI_SUPPORT= 1 -DMPI_HOME=/usr/local/openmpi -DNVSHMEM_SHMEM_SUPPORT=1 -DSHMEM_HOME=/usr/local/openmpi /var/tmp/nvshmem_src_2.9.0-2 && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target all -- -j$(nproc) && \
     cmake --build /var/tmp/nvshmem_src_2.9.0-2/build --target install -- -j$(nproc) && \
     rm -rf /var/tmp/nvshmem_src_2.9.0-2 /var/tmp/nvshmem_src_2.9.0-2.txz