def definition_func(
in_field: gtscript.Field[test_dtype],
out_field: gtscript.Field[test_dtype],
param: test_dtype,
):
with computation(PARALLEL), interval(...):
out_field = in_field + param
def definition_func(
in_field: gtscript.Field[float],
out_field: gtscript.Field[float],
*,
parameter: float = 1.0,
):
from gt4py.__gtscript__ import computation, interval, PARALLEL, FORWARD, BACKWARD
with computation(PARALLEL), interval(...):
out_field = in_field + parameter
with computation(FORWARD), interval(0, None):
out_field = in_field + parameter + 2.0
with computation(BACKWARD), interval(1, -1):
out_field = in_field + parameter + 2.0
with computation(FORWARD):
with interval(...):
out_field = in_field + parameter + 2.0
with computation(BACKWARD):
with interval(0, None):
out_field = in_field + parameter + 2.0
with computation(PARALLEL):
with interval(1, -1):
out_field = in_field + parameter + 2.0
with computation(BACKWARD):
with interval(...):
out_field = in_field + parameter + 2.0
with interval(...):
out_field = in_field + parameter + 2.0
with interval(...):
out_field = in_field + parameter + 2.0
with computation(PARALLEL):
with interval(0, None):
out_field = in_field + parameter + 2.0
with interval(...):
out_field = in_field + parameter + 2.0
with interval(1, -1):
out_field = in_field + parameter + 2.0
with computation(FORWARD):
with interval(1, -1):
out_field = in_field + parameter + 2.0
with interval(0, None):
out_field = in_field + parameter + 2.0
with interval(...):
out_field = in_field + parameter + 2.0
def definition_func(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import PARALLEL, computation, interval
with computation(PARALLEL), interval(...):
inout_field = ((inout_field[0, 0, 0] + GLOBAL_CONSTANT +
GLOBAL_NESTED_CONSTANTS.A +
GLOBAL_VERY_NESTED_CONSTANTS.nested.A)
if GLOBAL_BOOL_CONSTANT else 0)
def definition_func(inout_field: gtscript.Field[float]):
from __gtscript__ import computation, interval, PARALLEL, FORWARD, BACKWARD
from __externals__ import EXTERNAL
from gt4py.__gtscript__ import computation, interval, PARALLEL, FORWARD, BACKWARD
from gt4py.__externals__ import EXTERNAL
with computation(PARALLEL), interval(...):
inout_field = inout_field[0, 0, 0] + EXTERNAL
def definition_bw(in_field: gtscript.Field[np.float_],
out_field: gtscript.Field[np.float_]):
from gt4py.__gtscript__ import FORWARD, computation, interval
with computation(BACKWARD):
with interval(1, 2):
in_field = out_field + 1
with interval(0, 1):
in_field = out_field + 2
def sumdiff_defs(
in_a: gtscript.Field["dtype_in"],
in_b: gtscript.Field["dtype_in"],
out_c: gtscript.Field["dtype_out"],
out_d: gtscript.Field[float],
*,
wa: "dtype_scalar",
wb: int,
):
with computation(PARALLEL), interval(...):
out_c = wa * in_a + wb * in_b
out_d = wa * in_a - wb * in_b
def definition_func(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import computation, interval, PARALLEL
from gt4py.__externals__ import (
BOOL_CONSTANT,
CONSTANT,
NESTED_CONSTANTS,
VERY_NESTED_CONSTANTS,
)
with computation(PARALLEL), interval(...):
inout_field = ((inout_field[0, 0, 0] + CONSTANT +
NESTED_CONSTANTS.A +
VERY_NESTED_CONSTANTS.nested.A)
if GLOBAL_BOOL_CONSTANT else 0)
def definition_func(
in_phi: gtscript.Field[np.float64],
in_gamma: gtscript.Field[np.float64],
out_phi: gtscript.Field[np.float64],
out_field: gtscript.Field[np.float64],
*,
dx: float,
dy: float,
):
from gt4py.__gtscript__ import computation, interval, PARALLEL, FORWARD, BACKWARD
from gt4py.__externals__ import stage_laplacian, stage_laplacian_x, stage_laplacian_y
with computation(PARALLEL), interval(...):
lap = stage_laplacian(dx=dx, dy=dy, phi=in_phi) + GLOBAL_CONSTANT
out_phi = in_gamma[0, 0, 0] * lap[0, 0, 0]
with computation(PARALLEL), interval(...):
tmp_out = identity(in_phi)
out_phi = tmp_out + 1
with computation(PARALLEL), interval(...):
tmp_out2 = identity(in_gamma)
out_field = out_phi + tmp_out2
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(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import computation, interval, PARALLEL
from gt4py.__externals__ import WRONG_VALUE_CONSTANT
with computation(PARALLEL), interval(...):
inout_field = inout_field[0, 0, 0] + WRONG_VALUE_CONSTANT
def func(in_field: gtscript.Field[np.float_],
out_field: gtscript.Field[np.float_]):
from gt4py.__externals__ import offset
with computation(PARALLEL), interval(...):
out_field[0, 0, offset] = in_field
def definition_func(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import computation, interval, PARALLEL
from gt4py.__externals__ import NESTED_CONSTANTS
with computation(PARALLEL), interval(...):
inout_field = inout_field[0, 0, 0] + NESTED_CONSTANTS.missing
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 func(in_field: gtscript.Field[np.float_],
out_field: gtscript.Field[np.float_]):
with computation(PARALLEL), interval(...):
out_field["a_key"] = in_field
def func(in_field: gtscript.Field[np.float_]):
with computation(PARALLEL), interval(...):
in_field = not isfinite(in_field)
def definition_func(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import PARALLEL, computation, interval
with computation(PARALLEL), interval(...):
inout_field = inout_field[0, 0,
0] + GLOBAL_NESTED_CONSTANTS.missing
def func(in_field: gtscript.Field[np.float_],
out_field: gtscript.Field[np.float_]):
with computation(PARALLEL), interval(...):
tmp[0, 0, 0] = 2 * in_field
out_field = tmp
def func(in_field: gtscript.Field[np.float_]):
with computation(PARALLEL), interval(...):
in_field += 2.0
in_field -= 0.5
in_field /= 0.5
in_field *= 4.0
def definition(in_field: gtscript.Field[np.float_],
out_field: gtscript.Field[np.float_]):
with computation(PARALLEL):
with interval(...):
in_field = out_field
def definition_func(inout_field: gtscript.Field[float]):
from gt4py.__gtscript__ import computation, interval, PARALLEL
with computation(PARALLEL), interval(...):
inout_field = inout_field[0, 0, 0] + MISSING_CONSTANT
def func(in_field: gtscript.Field[np.float_]):
with computation(PARALLEL), interval(...):
in_field += sinus(in_field)
