(mod, target, exec_mode="bytecode")
| 89 | |
| 90 | |
| 91 | def codegen(mod, target, exec_mode="bytecode"): |
| 92 | builder = relax.ExecBuilder() |
| 93 | leftover_mod = relax.vm_build._vmcodegen(builder, mod, exec_mode=exec_mode) |
| 94 | tir_mod = relax.vm_build._filter_tir(leftover_mod) |
| 95 | return relax.vm_build._vmlink(builder, target, tir_mod) |
| 96 | |
| 97 | |
| 98 | @pytest.mark.gpu |
no outgoing calls
no test coverage detected
searching dependent graphs…