Support blocked parent storage in fixed patch stepping

2026-10-15 16:25:26 +08:00
parent ca5a145107
commit 906e0d9a0e
6 changed files with 496 additions and 66 deletions
@@ -128,8 +128,8 @@ class BlockedFineCPU(unittest.TestCase):
            backend = FixedPatchStepper(config("periodic"), BOUNDS, context=context,
                                        fine_block=FINE_BLOCKS[0])
            self.assertEqual(len(backend.fine_layout.halo_dst), 0)
-            self.assertIsNone(backend._halo_dst)
-            backend._fine_halo(Mock())  # no launch without pairs
+            self.assertIsNone(backend._levels[1].halo_dst)
+            backend._gather_halo(backend._levels[1], Mock())  # no launch without pairs
            module.get_function.return_value.assert_not_called()
            backend.close()
        self.assertTrue(sizes)
@@ -187,7 +187,7 @@ class BlockedFineCPU(unittest.TestCase):
            backend, calls = self.mocked_backend([circle(spec)])
            c, f = backend._levels
            coarse, cur, spare = c.current, f.current, f.spare
-            flags, kind = backend._fine_flags, backend._kind
+            flags, kind = backend._fine_flags, backend._levels[1].kind
            velocity = backend._link_tables.get("surface_velocity")
            self.assertEqual(velocity is None, spec is None)
            backend.step(1)