22#
33# SPDX-License-Identifier: (Apache-2.0 OR MIT)
44
5- from spack_repo .builtin .build_systems .cmake import CmakePackage
5+ from spack_repo .builtin .build_systems import cmake
6+ from spack_repo .builtin .build_systems .cmake import CMakePackage , generator
67from spack_repo .builtin .build_systems .rocm import ROCmPackage
78from spack_repo .builtin .build_systems .cuda import CudaPackage
89
910from spack .package import *
1011
11- class Tmlqcd (CmakePackage , CudaPackage , ROCmPackage ):
12- """Base class for building tmlQCD."""
12+
13+ class Tmlqcd (CMakePackage , CudaPackage , ROCmPackage ):
14+ """Base class for building tmlQCD."""
1315
1416 homepage = "https://www.itkp.uni-bonn.de/~urbach/software.html"
1517 url = "https://github.com/etmc/tmLQCD/archive/refs/tags/rel-5-1-6.tar.gz"
@@ -47,8 +49,8 @@ class Tmlqcd(CmakePackage, CudaPackage, ROCmPackage):
4749 variant ("half_spinor" , default = True , description = "Use a Dirac operator with half-spinor" )
4850 variant ("shared" , default = False , description = "Enable shared library" )
4951 variant ("shmem" , default = False , description = "Use shmem API" )
50- variant ("quda" , default = True , description = "Enable the QUDA library" , when = "+cuda" , )
51- variant ("quda" , default = True , description = "Enable the QUDA library" , when = "+rocm" , )
52+ variant ("quda" , default = True , description = "Enable the QUDA library" , when = "+cuda" )
53+ variant ("quda" , default = True , description = "Enable the QUDA library" , when = "+rocm" )
5254 variant (
5355 "QPhiX" , default = False , description = "Enable the QPhiX library for Intel Xeon and Xeon Phis"
5456 )
@@ -77,7 +79,7 @@ class Tmlqcd(CmakePackage, CudaPackage, ROCmPackage):
7779 depends_on ("lapack" )
7880 depends_on ("pkgconfig" , type = "build" )
7981
80- # dependencies
82+ # dependencies
8183 depends_on ("mpi" , when = "+mpi" )
8284 depends_on ("lemon-io" , when = "+lemon" )
8385
@@ -93,6 +95,7 @@ class Tmlqcd(CmakePackage, CudaPackage, ROCmPackage):
9395
9496 depends_on ("fftw-api@3" , when = "+fftw" )
9597
98+
9699class CMakeBuilder (cmake .CMakeBuilder ):
97100 def cmake_args (self ):
98101 spec = self .spec
@@ -101,7 +104,7 @@ def cmake_args(self):
101104 self .define_from_variant ("TM_USE_LEMON" , "lemon" ),
102105 self .define_from_variant ("TM_USE_MPI" , "mpi" ),
103106 self .define_from_variant ("TM_USE_QUDA" , "quda" ),
104- self .define_from_variant ("TM_USE_CUDA" ,"cuda" ),
107+ self .define_from_variant ("TM_USE_CUDA" , "cuda" ),
105108 self .define_from_variant ("TM_USE_HIP" , "cuda" ),
106109 self .define_from_variant ("TM_USE_FFTW" , "fftw" ),
107110 self .define_from_variant ("TM_FIXEDVOLUME" , "fixed_volume" ),
0 commit comments