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
