6b5694f0f2941fb9e8fa510011e3caeb03056b4e,tests/python/relay/test_simplify_fc_transpose.py,,run_func,#Any#Any#Any#,30

Before Change

def run_func(func, params, x):
    with tvm.transform.PassContext(opt_level=3):
        graph, lib, new_params = relay.build(func, "llvm", params=params)

    from tvm.contrib import graph_runtime

    ctx = tvm.cpu(0)
    dtype = "float32"
    m = graph_runtime.create(graph, lib, ctx)
    // set inputs
    m.set_input("data", tvm.nd.array(x.astype(dtype)))
    m.set_input(**new_params)
    // execute
    m.run()
    // get outputs
    tvm_output = m.get_output(0)

After Change

    ctx = tvm.cpu(0)
    dtype = "float32"
    m = graph_runtime.GraphModule(lib["default"](ctx))
    // set inputs
    m.set_input("data", tvm.nd.array(x.astype(dtype)))
    // execute
    m.run()

Instances