From 946dc7b16a7b53b2bab5264911063228e1a1c244 Mon Sep 17 00:00:00 2001
From: Sreenivasa Murthy Kolam <67086238+srekolam@users.noreply.github.com>
Date: Thu, 6 May 2021 02:42:37 -0400
Subject: Enable cmake option -optimal for rocsolver (#23448)

---
 var/spack/repos/builtin/packages/rocsolver/package.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'var')

diff --git a/var/spack/repos/builtin/packages/rocsolver/package.py b/var/spack/repos/builtin/packages/rocsolver/package.py
index 8d3b5f2e50..c89a38d949 100644
--- a/var/spack/repos/builtin/packages/rocsolver/package.py
+++ b/var/spack/repos/builtin/packages/rocsolver/package.py
@@ -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 @@ class Rocsolver(CMakePackage):
         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):
-- 
cgit v1.2.3-70-g09d2