diff --git a/experiments/amg2023/experiment.py b/experiments/amg2023/experiment.py index d4ef26666..5866eea6e 100644 --- a/experiments/amg2023/experiment.py +++ b/experiments/amg2023/experiment.py @@ -140,22 +140,21 @@ def compute_spack_section(self): system_specs = {} system_specs["compiler"] = "default-compiler" system_specs["mpi"] = "default-mpi" - system_specs["lapack"] = "lapack" + system_specs["lapack"] = "default-lapack" + system_specs["blas"] = "default-blas" + + # set package spack specs + # empty package_specs value implies external package + self.add_spack_spec(system_specs["mpi"]) + if self.spec.satisfies("+cuda"): system_specs["cuda_version"] = "{default_cuda_version}" system_specs["cuda_arch"] = "{cuda_arch}" - system_specs["blas"] = "cublas-cuda" - if self.spec.satisfies("+rocm"): + elif self.spec.satisfies("+rocm"): system_specs["rocm_arch"] = "{rocm_arch}" - system_specs["blas"] = "blas-rocm" - # set package spack specs - if self.spec.satisfies("+cuda") or self.spec.satisfies("+rocm"): - # empty package_specs value implies external package - self.add_spack_spec(system_specs["blas"]) - # empty package_specs value implies external package - self.add_spack_spec(system_specs["mpi"]) # empty package_specs value implies external package + self.add_spack_spec(system_specs["blas"]) self.add_spack_spec(system_specs["lapack"]) self.add_spack_spec(