def add_diffusive(fx: sd, fx2: sd, fy: sd, fy2: sd): with computation(PARALLEL), interval(...): fx[0, 0, 0] = fx + fx2 fy[0, 0, 0] = fy + fy2
def definition(inout_field: gtscript.Field[float]): with computation(PARALLEL), interval(...): compile_assert(GLOBAL_VERY_NESTED_CONSTANTS.nested.A > 1) inout_field = inout_field[0, 0, 0] + GLOBAL_VERY_NESTED_CONSTANTS.nested.A
def definition(inout_field: gtscript.Field[float]): with computation(PARALLEL), interval(...): compile_assert(inout_field[0, 0, 0] < 0)
def definition_func(field: gtscript.Field[float]): with computation(FORWARD): with interval(0, 3): field = 0 with interval(2, None): field = 1
def definition_func(phi: gtscript.Field[np.float64]): with computation(PARALLEL), interval(...): phi = test_conditional_return(phi)
def definition_func(inout_field: gtscript.Field[float]): from gt4py.__externals__ import NESTED_CONSTANTS from gt4py.__gtscript__ import PARALLEL, computation, interval with computation(PARALLEL), interval(...): inout_field = inout_field[0, 0, 0] + NESTED_CONSTANTS.missing
def definition_func(field: gtscript.Field[float]): from gt4py.__externals__ import kstart with computation(PARALLEL), interval(kstart, -1): field = 0
def definition(field_a, *, scale): with computation(PARALLEL), interval(...): field_a = scale * field_a
def definition(field_a): with computation(PARALLEL), interval(...): tmp = field_a field_a = tmp
def set_dp(dp1: FloatField, pe1: FloatField): with computation(PARALLEL), interval(...): dp1 = pe1[0, 0, 1] - pe1
def definition(field_a): from __externals__ import SCALE_FACTOR with computation(PARALLEL), interval(...): field_a = SCALE_FACTOR * field_a[0, 0, 0]
def diffusive_damp_y(fy: sd, fy2: sd, mass: sd, damp: float): with computation(PARALLEL), interval(...): fy[0, 0, 0] = fy + 0.5 * damp * (mass[0, -1, 0] + mass) * fy2
def diffusive_damp_x(fx: sd, fx2: sd, mass: sd, damp: float): with computation(PARALLEL), interval(...): fx = fx + 0.5 * damp * (mass[-1, 0, 0] + mass) * fx2
def add_diffusive_component(fx: sd, fx2: sd): with computation(PARALLEL), interval(...): fx[0, 0, 0] = fx + fx2
def definition_func(inout_field: gtscript.Field[float]): from gt4py.__externals__ import some_call with computation(PARALLEL), interval(...): inout_field = func() + some_call()
def definition(infield, outfield): with computation(PARALLEL), interval(...): outfield = ( # noqa: F841 # Local name is assigned to but never used infield > 0.0) * infield + (infield <= 0.0) * (-infield[0, 1, 0])
def definition_func(inout_field: gtscript.Field[float]): from gt4py.__gtscript__ import PARALLEL, computation, interval with computation(PARALLEL), interval(...): inout_field = func()
def definition(field_a, field_b): with computation(PARALLEL), interval(...): field_b = field_a # noqa: F841 # Local name is assigned to but never used
def definition_func(inout_field: gtscript.Field[float]): from gt4py.__externals__ import WRONG_VALUE_CONSTANT with computation(PARALLEL), interval(...): inout_field = inout_field[0, 0, 0] + WRONG_VALUE_CONSTANT
def definition(field_a, field_b): with computation(PARALLEL), interval(...): field_a += 1.0 field_a *= 2.0 field_b -= 1.0 field_b /= 2.0
def definition_func(field: gtscript.Field[float]): with computation(PARALLEL), interval(-1, 1): field = 0
def func(in_field: gtscript.Field[np.float_]): with computation(PARALLEL), interval(...): in_field += min(abs(sin(add_external_const(in_field))), -0.5)
def definition_func(phi: gtscript.Field[np.float64], dx: float): from gt4py.__externals__ import lap with computation(PARALLEL), interval(...): phi = lap(lap(phi, dx), dx)
def func(in_field: gtscript.Field[np.float_]): with computation(PARALLEL), interval(...): in_field += sinus(in_field)
def definition(inout_field: gtscript.Field[float]): from gt4py.__externals__ import EXTERNAL with computation(PARALLEL), interval(...): compile_assert(EXTERNAL < 1) inout_field = inout_field[0, 0, 0] + EXTERNAL
def func(in_field: gtscript.Field[np.float_]): with computation(PARALLEL), interval(...): in_field = not isfinite(in_field)
def definition(inout_field: gtscript.Field[float]): with computation(PARALLEL), interval(...): inout_field = assert_in_func(inout_field)
def func(in_field: gtscript.Field[np.float_]): with computation(PARALLEL), interval(...): in_field = asin(in_field) + 1 if 1 < in_field else sin(in_field)
def definition( field_in: gtscript.Field[gtscript.IJK, np.float_], field_out: gtscript.Field[gtscript.K, np.float_], ): with computation(PARALLEL), interval(...): field_out = field_in[0, 0, 0]
def d2_damp(q: sd, d2: sd, damp: float): with computation(PARALLEL), interval(...): d2[0, 0, 0] = damp * q