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)