Function test_add

tests/python/relax/test_codegen_coreml.py:93–105

()

@requires_coreml_runtime
def test_add():
    x = relax.Var("x", relax.TensorStructInfo([10, 10], "float32"))
    y = relax.Var("y", relax.TensorStructInfo([10, 10], "float32"))
    bb = relax.BlockBuilder()
    with bb.function("main", [x, y]):
        with bb.dataflow():
            lv0 = bb.emit(relax.op.add(x, y))
            gv = bb.emit_output(lv0)
        bb.emit_func_output(gv)
    mod = bb.get()
    x_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev)
    y_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev)
    verify(mod, [x_data, y_data])


@requires_coreml_runtime