* Eric Irrgang <ericirrgang@gmail.com>
* Joe Jordan <e.jjordan12@gmail.com>
* Mark Abraham <mark.j.abraham@gmail.com>
+ * Gaurav Garg <gaugarg@nvidia.com>
"""
Instead, inherit from it with the *parents* argument to :py:class:`argparse.ArgumentParser`
"""
-parser.add_argument('--cmake', nargs='*', type=str, default=['3.16.3', '3.17.2', '3.18.4'], # new minimum required versions
+parser.add_argument('--cmake', nargs='*', type=str, default=['3.16.3', '3.17.2', '3.18.4', '3.21.2'], # new minimum required versions
help='Selection of CMake version to provide to base image')
compiler_group = parser.add_mutually_exclusive_group()
parser.add_argument('--rocm', type=str, nargs='?', const='debian', default=None,
help='Select AMD compute engine version.')
-parser.add_argument('--intel-compute-runtime', type=str, nargs='?', default=None,
- help='Select Intel Compute Runtime version.')
+parser.add_argument('--intel-compute-runtime', action='store_true', default=False,
+ help='Include Intel Compute Runtime.')
parser.add_argument('--clfft', type=str, nargs='?', const='master', default=None,
help='Add external clFFT libraries to the build image')
+parser.add_argument('--heffte', type=str, nargs='?', default=None,
+ help='Select heffte repository tag/commit/branch.')
+
parser.add_argument('--doxygen', type=str, nargs='?', const='1.8.5', default=None,
help='Add doxygen environment for documentation builds. Also adds other requirements needed for final docs images.')
elements.append(gpusdk + '-' + version)
if configuration.oneapi is not None:
elements.append('oneapi-' + configuration.oneapi)
- if configuration.intel_compute_runtime is not None:
- elements.append('intel-' + configuration.intel_compute_runtime)
+ if configuration.intel_compute_runtime:
+ elements.append('intel-compute-runtime')
if configuration.rocm is not None:
if (configuration.rocm != 'debian'):
elements.append('rocm-' + configuration.rocm)