Function test_add_const

tests/python/relax/test_codegen_coreml.py:109–120

()

@requires_coreml_runtime
def test_add_const():
    x = relax.Var("x", relax.TensorStructInfo([10, 10], "float32"))
    y = relax.const(np.ones([10, 10]), "float32")
    bb = relax.BlockBuilder()
    with bb.function("main", [x]):
        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)
    verify(mod, [x_data])


@requires_coreml_runtime