summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
authorTodd Gamblin <tgamblin@llnl.gov>2019-07-07 23:14:04 -0700
committerTodd Gamblin <tgamblin@llnl.gov>2020-11-17 10:04:13 -0800
commitc7812f7e10d08f1f9b3ecd66fc4ed0edd79d2ba5 (patch)
treebd5e33249763b35f2cfa1be425ed6ee8022f7774 /lib
parenta8a6d943d68deafbff6dd672b92404cf00a89e2b (diff)
downloadspack-c7812f7e10d08f1f9b3ecd66fc4ed0edd79d2ba5.tar.gz
spack-c7812f7e10d08f1f9b3ecd66fc4ed0edd79d2ba5.tar.bz2
spack-c7812f7e10d08f1f9b3ecd66fc4ed0edd79d2ba5.tar.xz
spack-c7812f7e10d08f1f9b3ecd66fc4ed0edd79d2ba5.zip
concretizer: add rudimentary variants with defaults to ASP solve
Diffstat (limited to 'lib')
-rw-r--r--lib/spack/spack/solver/asp.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/lib/spack/spack/solver/asp.py b/lib/spack/spack/solver/asp.py
index f810e5a1a9..3c9bdd15d1 100644
--- a/lib/spack/spack/solver/asp.py
+++ b/lib/spack/spack/solver/asp.py
@@ -165,6 +165,24 @@ def pkg_rules(pkg):
# versions
pkg_version_rules(pkg)
+ # variants
+ for name, variant in pkg.variants.items():
+ asp._rule(asp.variant(pkg.name, name),
+ asp.node(pkg.name))
+
+ single_value = not variant.multi
+ single = asp.variant_single_value(pkg.name, name)
+ if single_value:
+ asp._rule(single, asp.node(pkg.name))
+ asp._rule(asp.variant_value(pkg.name, name, variant.default),
+ asp._not(asp.variant_set(pkg.name, name)))
+ else:
+ asp._rule(asp._not(single), asp.node(pkg.name))
+ defaults = variant.default.split(',')
+ for val in defaults:
+ asp._rule(asp.variant_value(pkg.name, name, val),
+ asp._not(asp.variant_set(pkg.name, name)))
+
# dependencies
for name, conditions in pkg.dependencies.items():
for cond, dep in conditions.items():
@@ -210,6 +228,10 @@ _generate = """\
{ arch_platform(P, A) : arch_platform(P, A) } = 1 :- node(P).
{ arch_os(P, A) : arch_os(P, A) } = 1 :- node(P).
{ arch_target(P, T) : arch_target(P, T) } = 1 :- node(P).
+
+% one variant value for single-valued variants.
+{ variant_value(P, V, X) : variant_value(P, V, X) } = 1
+ :- node(P), variant(P, V), not variant_single_value(P, V).
"""
#: define the rules of Spack concretization
@@ -221,6 +243,13 @@ node(D) :- node(P), depends_on(P, D).
arch_platform(D, A) :- node(D), depends_on(P, D), arch_platform(P, A).
arch_os(D, A) :- node(D), depends_on(P, D), arch_os(P, A).
arch_target(D, A) :- node(D), depends_on(P, D), arch_target(P, A).
+
+% if a variant is set to anything, it is considered "set".
+variant_set(P, V) :- variant_set(P, V, _).
+
+% variant_set is an explicitly set variant value. If it's not "set",
+% we revert to the default value. If it is set, we force the set value
+variant_value(P, V, X) :- node(P), variant(P, V), variant_set(P, V, X).
"""
#: what parts of the model to display to read models back in
@@ -228,6 +257,7 @@ _display = """\
#show node/1.
#show depends_on/2.
#show version/2.
+#show variant_value/3.
#show arch_platform/2.
#show arch_os/2.
#show arch_target/2.