Python NativeFuncCall示例

示例#1

文件： gtir_to_oir.py 项目： fthaler/gt4py

 def visit_NativeFuncCall(self, node: gtir.NativeFuncCall) -> oir.NativeFuncCall:
     return oir.NativeFuncCall(
         func=node.func,
         args=self.visit(node.args),
         dtype=node.dtype,
         kind=node.kind,
         loc=node.loc,
     )

示例#2

文件： test_oir_to_npir.py 项目： eddie-c-davis/gt4py

def test_native_func_call():
    oir_node = oir.NativeFuncCall(
        func=common.NativeFunction.SQRT,
        args=[
            oir.FieldAccess(
                name="a",
                offset=common.CartesianOffset.zero(),
                dtype=common.DataType.FLOAT64,
            ),
        ],
    )
    result = OirToNpir().visit(
        oir_node,
        parallel_k=True,
        ctx=OirToNpir.ComputationContext(),
    )
    assert isinstance(result, npir.VectorExpression)

示例#3

 def visit_NativeFuncCall(self, node: gtir.NativeFuncCall, **kwargs: Any) -> oir.NativeFuncCall:
     return oir.NativeFuncCall(
         func=node.func, args=self.visit(node.args), dtype=node.dtype, kind=node.kind
     )