Fix amg default blas lapack config (#530)

* Fix amg default blas lapack config * Fix system configs --------- Co-authored-by: Riyaz Haque <[email protected]> Co-authored-by: pearce8 <[email protected]>
LLNL · Jan 8, 2025 · 8a935ea · 8a935ea
1 parent 87dd631
commit 8a935ea
Showing 1 changed file with 9 additions and 10 deletions.
diff --git 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(