Lbann package: bug fix (#7066)
Fixed a bug in the guard on cub functionality. Reduced the requirements on hwloc to make it simpler to build on Power systems.
This commit is contained in:
parent
85c887a939
commit
82d4d58ca7
@ -56,7 +56,7 @@ class Lbann(CMakePackage):
|
||||
depends_on('cudnn', when='+gpu')
|
||||
depends_on('cub', when='+gpu')
|
||||
depends_on('mpi')
|
||||
depends_on('hwloc')
|
||||
depends_on('hwloc ~pci ~libxml2')
|
||||
depends_on('opencv@3.2.0: +openmp +core +highgui +imgproc +jpeg +png +tiff +zlib ~eigen', when='+opencv')
|
||||
depends_on('protobuf@3.0.2:')
|
||||
depends_on('cnpy')
|
||||
@ -148,7 +148,7 @@ def cmake_args(self):
|
||||
args.extend(['-DcuDNN_DIR={0}'.format(
|
||||
spec['cudnn'].prefix)])
|
||||
|
||||
if '+cudnn' in spec:
|
||||
if '+cub' in spec:
|
||||
args.extend(['-DCUB_DIR={0}'.format(
|
||||
spec['cub'].prefix)])
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user