(
vm: relax.VirtualMachine, device: tvm.runtime.Device
)
| 1202 | |
| 1203 | |
| 1204 | def save_function_time_evaluator_trial( |
| 1205 | vm: relax.VirtualMachine, device: tvm.runtime.Device |
| 1206 | ) -> None: |
| 1207 | # just checking that the saved function can be called in the time evaluator |
| 1208 | a = tvm.runtime.tensor(np.random.rand(32, 32).astype("float32"), device) |
| 1209 | b = tvm.runtime.tensor(np.random.rand(32, 32).astype("float32"), device) |
| 1210 | vm.save_function("main", "saved_main", a, b) |
| 1211 | vm.time_evaluator("saved_main", device)() |
| 1212 | |
| 1213 | |
| 1214 | def test_save_function_time_evaluator(exec_mode): |
no test coverage detected
searching dependent graphs…