ClaudeandClaude Opus 5.5
2b10b01c49
Retired with the dense refinement route (results recorded at1573942and2469af7or named as accepted losses): - test_refinement_blocked_base: BlockedBaseGPU exact parity (3 tests), DenseBlockedRouteParity (6 tests incl. MRT and two bodies), DenseBlockedRouteCPU (capability matrix, ledger comparator); - test_refinement_blocked_fine: BlockedFineGPU dense-vs-blocked (3 tests); - test_refinement_mrt_gpu: test_dense_versus_blocked_mrt_two_level; - test_raw_exchange_gpu: test_dense_fine_storage_reference (private link_exchange cut-link oracle, accepted loss); - test_refinement_gpu: test_bounded_equivalence_and_ownership (dense CELERIS_REFINEMENT_AUDIT ownership counters, accepted loss); its physics stays in test_bounded_equivalence_blocked; eight_steps loses the audit branch and the misleading 'ownership/counts PASS' label; - test_fine_surface_gpu: test_fine_surface_four_hops (dense per-hop flux ledger, CurvedLinkSoA reflection oracle, audit counts: accepted losses; raw exchange covers the boundary part only, not full flux closure) and the dense SoA fault tests (missing surface buffer, close frees SoA); blocked equivalents remain in test_refinement_blocked_fine/_base; - test_raw_exchange.test_dense_observation_rejected_before_device_access. New TileContractCPU: level_blocks omitted/None/wrong length/None entry/ wrong arity/non-integer/bool/non-positive/non-tuple/not dividing all raise before CUDA; whole-level tiles reach CUDA. Rejection tests and fixtures pass tiles; launch-order and kernel-text tests follow the blocked root; examples/fixed_periodic_patch uses whole-level tiles (+ host tile test). Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
78 lines
4.3 KiB
Python
78 lines
4.3 KiB
Python
"""MRT reuse GPU gates. Caller owns GPU admission (CUDA_VISIBLE_DEVICES).
|
|
|
|
Run: python -m unittest discover -s tests/validation/multigrid -p test_refinement_mrt_gpu.py
|
|
Parity compares the blocked MRT channel root with uniform MRT OneStep: the existing
|
|
pressure-NEQ outlet branch differs by collision model, so SRT is never the reference.
|
|
AMR checks are finite-evolution/split-invariance smoke tests, not benchmark validation.
|
|
"""
|
|
from dataclasses import replace
|
|
import unittest
|
|
import numpy as np
|
|
|
|
from CelerisLab.config import LBMConfig
|
|
from CelerisLab.cuda.context import CudaContext
|
|
from CelerisLab.lbm.population_storage import _encode_physical
|
|
from CelerisLab.lbm.refinement_stepper import FixedPatchStepper
|
|
from test_channel_root_gpu import run_root, state
|
|
from test_refinement_blocked_base import raw, exact
|
|
|
|
|
|
class RefinementMRTGPU(unittest.TestCase):
|
|
def test_uniform_matched_mrt_root(self):
|
|
with CudaContext(0) as context:
|
|
for shifted in (False, True):
|
|
for profile in ("uniform", "parabolic"):
|
|
cfg = LBMConfig(nx=24, ny=20, viscosity=.1, velocity=.025, collision="MRT",
|
|
ddf_shifting=shifted, inlet_profile=profile)
|
|
initial = _encode_physical(state(cfg.ny, cfg.nx), shifted)
|
|
expected = run_root(cfg, context, initial)
|
|
self.assertTrue(np.isfinite(expected).all())
|
|
for block in ((20, 24), (5, 6), (4, 1)):
|
|
np.testing.assert_array_equal(run_root(cfg, context, initial, block), expected)
|
|
|
|
def test_amr_mrt_smoke(self):
|
|
from CelerisLab.body.geometry.circle import CircleGeometry
|
|
from CelerisLab.body.objects import SimObject
|
|
center = (24., 16.)
|
|
body = SimObject(0, geometry=CircleGeometry(*center, 2.15), center=center, radius=2.15)
|
|
with CudaContext(0) as context:
|
|
for boundary in ("periodic", "channel"):
|
|
for shifted in (False, True):
|
|
cfg = LBMConfig(nx=48, ny=32, viscosity=.1, velocity=.015 if boundary == "channel" else 0.,
|
|
collision="MRT", ddf_shifting=shifted)
|
|
reference = None
|
|
for block in ((32, 48), (8, 8), (4, 1)):
|
|
with FixedPatchStepper(cfg, context=context, coarse_boundary=boundary,
|
|
refinements=[(8, 6, 40, 26)], level_blocks=[block, (8, 8)],
|
|
level_objects=[(), (body,)]) as backend:
|
|
backend.initialize(*[state(*level.shape[1:]) for level in backend._levels])
|
|
backend.step(8)
|
|
raw = backend._download_raw()
|
|
for a, mask in zip(raw.values, backend._state):
|
|
self.assertTrue(np.isfinite(a[:, mask]).all())
|
|
if reference is None: reference = raw.values
|
|
else:
|
|
for a, b in zip(reference, raw.values): np.testing.assert_array_equal(a, b)
|
|
|
|
def test_shifted_all_blocked_three_level_mrt_smoke(self):
|
|
"""Per-level shear rate omega_l = 1/(3*2**l*nu + 1/2); finite ACTIVE leaves."""
|
|
cfg = LBMConfig(nx=16, ny=16, viscosity=.1, collision="MRT", ddf_shifting=True)
|
|
kw = dict(refinements=[(4, 4, 12, 12), (8, 8, 16, 16)], level_blocks=[(4, 4)] * 3)
|
|
with CudaContext(0) as context:
|
|
with FixedPatchStepper(cfg, context=context, **kw) as backend:
|
|
self.assertEqual(len(backend._level_cfgs), 3)
|
|
for level, level_cfg in enumerate(backend._level_cfgs):
|
|
checked = replace(level_cfg)
|
|
checked.validate() # omega is derived only by validate()
|
|
self.assertEqual(checked.collision, "MRT")
|
|
self.assertTrue(checked.ddf_shifting)
|
|
self.assertTrue(np.isclose(checked.omega, 1 / (3 * 2**level * .1 + .5), rtol=1e-14))
|
|
backend.initialize(*[state(*level.shape[1:]) for level in backend._levels])
|
|
backend.step(4)
|
|
values = backend._download_raw().values
|
|
for a, mask in zip(values, backend._state):
|
|
self.assertTrue(np.isfinite(a[:, mask]).all())
|
|
|
|
|
|
if __name__ == "__main__": unittest.main()
|