icon: add dependency on cutensor

C2SM · Aug 20, 2024 · da976e4 · da976e4
1 parent 2de9b2b
commit da976e4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/repos/c2sm/packages/icon/package.py b/repos/c2sm/packages/icon/package.py
@@ -260,6 +260,7 @@ class Icon(AutotoolsPackage, CudaPackage):
     depends_on('yaxt+fortran', when='+cdi-pio')
     depends_on('lapack')
     depends_on('blas')
+    depends_on('cutensor')
     depends_on('netcdf-fortran')
 
     depends_on('netcdf-c', when='~cdi-pio')
@@ -449,7 +450,7 @@ def configure_args(self):
         elif self.compiler.name in ['pgi', 'nvhpc']:
             config_vars['CFLAGS'].extend(['-g', '-O2'])
             config_vars['FCFLAGS'].extend(
-                ['-g', '-O', '-Mrecursive', '-Mallocatable=03', '-Mbackslash'])
+                ['-g', '-O', '-Mrecursive', '-Mallocatable=03', '-Mbackslash', '-cudalib=cutensor'])
 
             if self.spec.variants['gpu'].value == 'openacc+cuda':
                 config_vars['FCFLAGS'].extend([
@@ -529,6 +530,7 @@ def configure_args(self):
 
         libs += self.spec['lapack:fortran'].libs
         libs += self.spec['blas:fortran'].libs
+        libs += self.spec['cutensor'].libs
         libs += self.spec['netcdf-fortran'].libs
 
         if '+coupling' in self.spec or '~cdi-pio' in self.spec: