diff options
author | Brian Van Essen <vanessen1@llnl.gov> | 2021-09-26 06:50:09 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2021-09-26 15:50:09 +0200 |
commit | cc658c5f1e065a206555668b4aaa49b568839279 (patch) | |
tree | 4f91da43fa9952f9330da63f4dc89e0f4049c4f7 /var | |
parent | 1f972f90c8fb3e7ec5b09382fc99508597b8ccbe (diff) | |
download | spack-cc658c5f1e065a206555668b4aaa49b568839279.tar.gz spack-cc658c5f1e065a206555668b4aaa49b568839279.tar.bz2 spack-cc658c5f1e065a206555668b4aaa49b568839279.tar.xz spack-cc658c5f1e065a206555668b4aaa49b568839279.zip |
lbann: add support for building with the ONNX C++ library (#26130)
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/lbann/package.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/lbann/package.py b/var/spack/repos/builtin/packages/lbann/package.py index 2966d51bf8..650351938e 100644 --- a/var/spack/repos/builtin/packages/lbann/package.py +++ b/var/spack/repos/builtin/packages/lbann/package.py @@ -68,6 +68,7 @@ class Lbann(CMakePackage, CudaPackage, ROCmPackage): description='Builds with support for image processing data with OpenCV') variant('vtune', default=False, description='Builds with support for Intel VTune') variant('onednn', default=False, description='Support for OneDNN') + variant('onnx', default=False, description='Support for exporting models into ONNX format') variant('nvshmem', default=False, description='Support for NVSHMEM') variant('python', default=True, description='Support for Python extensions (e.g. Data Reader)') variant('pfe', default=True, description='Python Frontend for generating and launching models') @@ -237,6 +238,7 @@ class Lbann(CMakePackage, CudaPackage, ROCmPackage): depends_on('llvm-openmp', when='%apple-clang') depends_on('onednn cpu_runtime=omp gpu_runtime=none', when='+onednn') + depends_on('onnx', when='+onnx') depends_on('nvshmem', when='+nvshmem') depends_on('zstr') @@ -307,6 +309,7 @@ class Lbann(CMakePackage, CudaPackage, ROCmPackage): '-DLBANN_WITH_NVSHMEM:BOOL=%s' % ('+nvshmem' in spec), '-DLBANN_WITH_FFT:BOOL=%s' % ('+fft' in spec), '-DLBANN_WITH_ONEDNN:BOOL=%s' % ('+onednn' in spec), + '-DLBANN_WITH_ONNX:BOOL=%s' % ('+onnx' in spec), '-DLBANN_WITH_EMBEDDED_PYTHON:BOOL=%s' % ('+python' in spec), '-DLBANN_WITH_PYTHON_FRONTEND:BOOL=%s' % ('+pfe' in spec), '-DLBANN_WITH_TBINF=OFF', |