diff options
| author | Brian Van Essen <vanessen1@llnl.gov> | 2021-01-27 12:11:31 -0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-01-27 21:11:31 +0100 |
| commit | 3cbc35dbdd335c7030d29b130484e83028570122 (patch) | |
| tree | 0ea70cc27a15cea4babab4bb08832a87d54e501c | |
| parent | aac36997487cb01ccb1f344fc0ccd87d0c577f5a (diff) | |
| download | spack-3cbc35dbdd335c7030d29b130484e83028570122.tar.gz spack-3cbc35dbdd335c7030d29b130484e83028570122.tar.bz2 spack-3cbc35dbdd335c7030d29b130484e83028570122.tar.xz spack-3cbc35dbdd335c7030d29b130484e83028570122.zip | |
LBANN: add support for building with OneDNN. (#21305)
| -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 b0ca29b709..7dee4809ae 100644 --- a/var/spack/repos/builtin/packages/lbann/package.py +++ b/var/spack/repos/builtin/packages/lbann/package.py @@ -62,6 +62,7 @@ class Lbann(CMakePackage, CudaPackage): variant('vision', default=False, 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('nvshmem', default=False, description='Support for NVSHMEM') # Variant Conflicts @@ -174,6 +175,7 @@ class Lbann(CMakePackage, CudaPackage): depends_on('llvm-openmp', when='%apple-clang') + depends_on('onednn cpu_runtime=omp gpu_runtime=none', when='+onednn') depends_on('nvshmem', when='+nvshmem') generator = 'Ninja' @@ -227,6 +229,7 @@ class Lbann(CMakePackage, CudaPackage): '-DLBANN_WITH_CUDNN:BOOL=%s' % ('+cuda' in spec), '-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_TBINF=OFF', '-DLBANN_WITH_UNIT_TESTING:BOOL=%s' % (self.run_tests), '-DLBANN_WITH_VISION:BOOL=%s' % ('+vision' in spec), |