uniform removal of sink sites

lehner · Jan 2, 2025 · daf7397 · daf7397
1 parent 9b051ba
commit daf7397
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 10 deletions.
diff --git a/lib/gpt/qcd/sparse_propagator.py b/lib/gpt/qcd/sparse_propagator.py
@@ -92,13 +92,8 @@ def __init__(self, array, cache_size, cache_line_size):
         self.cache_hits = 0
         self.cache_misses = 0
 
-        self.sink_domain_update()
-
-    def sink_domain_update(self):
+    def sink_domain_update(self, min_source_sampled_sites):
         self.cache = []
-        min_source_sampled_sites = min(
-            [flav.source_domain.sampled_sites for fac, flav in self.array]
-        )
         self.coordinates = self.sink_domain.coordinates
         self.ec = self.sink_domain.sdomain.unique_embedded_coordinates(
             self.coordinates[0:min_source_sampled_sites]
@@ -296,14 +291,21 @@ def flavor(roots, *cache_param):
 
         ret.append(prp)
 
-    # make common sink domain
+    # select common sink domain
     common_sink_domain = ret[0].sink_domain
     for flav in ret[1:]:
         common_sink_domain.restrict(flav.sink_domain)
+    max_ec_size = max(
+        [
+            min([flav.source_domain.sampled_sites for fac, flav in flav_mult.array])
+            for flav_mult in ret
+        ]
+    )
+
+    # prepare common sink domain
     for flav in ret:
-        if flav.sink_domain is not common_sink_domain:
-            flav.sink_domain = common_sink_domain
-            flav.sink_domain_update()
+        flav.sink_domain = common_sink_domain
+        flav.sink_domain_update(max_ec_size)
 
     if len(ret) == 1:
         ret = ret[0]

diff --git a/tests/qcd/sparse_propagator.py b/tests/qcd/sparse_propagator.py
@@ -129,6 +129,9 @@
     quark_l.sink_domain.sdomain.kernel.local_coordinates,
 )
 
+assert len(quark_s.coordinates) == len(quark_l.coordinates)
+assert len(quark_s.ec) == len(quark_l.ec)
+
 for i in range(quark_sml.source_domain.sampled_sites):
     qs_i = quark_s[i]
     ql_i = quark_l[i]