(mod, inputs, legalize=False)
| 52 | |
| 53 | |
| 54 | def build_and_run(mod, inputs, legalize=False): |
| 55 | target = tvm.target.Target("llvm") |
| 56 | dev = tvm.cpu() |
| 57 | inputs = [tvm.runtime.tensor(inp, dev) for inp in inputs] |
| 58 | |
| 59 | with tvm.transform.PassContext(config={"relax.transform.apply_legalize_ops": legalize}): |
| 60 | ex = tvm.compile(mod, target) |
| 61 | vm = relax.VirtualMachine(ex, dev) |
| 62 | f = vm["main"] |
| 63 | return f(*inputs).numpy() |
| 64 | |
| 65 | |
| 66 | def test_dnnl_offload(): |