gridtools-cpp

import gridtools import numpy as np # Define a simple stencil: compute average of neighbors def stencil(in_field, out_field): # GridTools expects Fortran-layout arrays in_field = np.asfortranarray(in_field) out_field = np.asfortranarray(out_field) # Example using the gridtools library (mock) # Real usage involves building a stencil object and applying it out_field[1:-1, 1:-1] = (in_field[:-2, 1:-1] + in_field[2:, 1:-1] + in_field[1:-1, :-2] + in_field[1:-1, 2:]) / 4.0 return out_field # Create test data nx, ny = 10, 10 in_field = np.random.rand(nx, ny) out_field = np.zeros((nx, ny)) result = stencil(in_field, out_field) print(result)