f143881314fdc0518a370e6353c6fb6305cedf23,tests/python/unittest/test_target_codegen_static_init.py,,test_static_init,#,54

Before Change

        return sh

    stmt = ib.get()
    fapi = tvm.testing.MakeAPILegacy(stmt, "ramp", [Ab], 0, True)
    f = tvm.driver.build(fapi, target="llvm")
    a = tvm.nd.array(np.zeros(10, dtype=dtype))
    f(a)

After Change

        return sh

    stmt = ib.get()
    mod = tvm.IRModule.from_expr(
        tvm.tir.PrimFunc([Ab], stmt).with_attr("global_symbol", "ramp"))
    f = tvm.driver.build(mod, target="llvm")
    a = tvm.nd.array(np.zeros(10, dtype=dtype))
    f(a)

Instances