zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enable cmake option -optimal for rocsolver (#23448)

Browse Source
This commit is contained in:
Sreenivasa Murthy Kolam 2021-05-06 02:42:37 -04:00 committed by GitHub
parent f9860de2d5
commit 946dc7b16a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
var/spack/repos/builtin/packages/rocsolver/package.py
View File

@ -16,6 +16,11 @@ class Rocsolver(CMakePackage):
maintainers = ['srekolam', 'arjun-raj-kuppala']
variant('optimal', default=True,
description='This option improves performance at the cost of increased binary \
size and compile time by adding specialized kernels \
for small matrix sizes')
version('4.1.0', sha256='da5cc800dabf7367b02b73c93780b2967f112bb45232e4b06e5fd07b4d5b8d88')
version('4.0.0', sha256='be9a52644c276813f76d78f2c11eddaf8c2d7f9dd04f4570f23d328ad30d5880')
version('3.10.0', sha256='bc72483656b6b23a1e321913a580ca460da3bc5976404647536a01857f178dd2')
@ -46,6 +51,9 @@ def cmake_args(self):
if self.spec.satisfies('@4.1.0:'):
args.append('-DCMAKE_CXX_FLAGS:String=-I{0}/rocblas/include'.format(incl))
if self.spec.satisfies('@3.7.0:'):
args.append(self.define_from_variant('OPTIMAL', 'optimal'))
return args
def setup_build_environment(self, env):