Function gen_ground_truth

tests/python/relax/test_transform_codegen_pass.py:78–89

(mod, target, dev, inputs)

def gen_ground_truth(mod, target, dev, inputs):
    # Lower and run tuning
    # Since there is no default schedule for GPU in MS yet, this is necessary
    with target:
        seq = tvm.transform.Sequential(
            [relax.transform.LegalizeOps(), s_tir.transform.DefaultGPUSchedule()]
        )
        new_mod = seq(mod)
    relax.analysis.well_formed(new_mod)
    exec = tvm.compile(new_mod, target, params={})
    vm = relax.VirtualMachine(exec, dev)
    return vm["main"](*inputs)


@tvm.script.ir_module