expose flow and hhd for benchmark

This commit is contained in:
radxa-d6a
2026-02-18 21:28:32 +08:00
parent ffb0305e10
commit 6a0c3f16ea
3 changed files with 19 additions and 11 deletions
+2 -2
View File
@@ -1628,7 +1628,7 @@ class PoissonSolver(object):
Batched OpenCL Poisson solver for two 2D fields.
Optimized: scatter H2D (~4x smaller transfer), fused multiply
(reads g_fft once), async D2H (one sync instead of two).
When profile=True, records and optionally prints per-stage timings.
When profile=False, records and optionally prints per-stage timings.
"""
if not OPENCL_AVAILABLE:
if verbose:
@@ -1818,7 +1818,7 @@ class PoissonSolver(object):
# """
# OpenCL Poisson solver using pyvkfft FFT convolution.
# Replaces scipy.signal.fftconvolve(..., mode='same') for structured grids.
# When profile=True, records and optionally prints per-stage timings.
# When profile=False, records and optionally prints per-stage timings.
# """
# if not OPENCL_AVAILABLE:
# if verbose: