()
| 106 | |
| 107 | |
| 108 | def setup_test(): |
| 109 | # Prepare IRModule and its input |
| 110 | mod = InputModule |
| 111 | assert isinstance(mod, tvm.IRModule) |
| 112 | |
| 113 | np0 = np.random.rand(16, 16).astype(np.float32) |
| 114 | np1 = np.random.rand(16, 16).astype(np.float32) |
| 115 | data0 = tvm.runtime.tensor(np0, dev) |
| 116 | data1 = tvm.runtime.tensor(np1, dev) |
| 117 | inputs = [data0, data1] |
| 118 | |
| 119 | # Ground truth should be generated before annotation |
| 120 | # due to the conflict with MS task extraction |
| 121 | # TODO(@sunggg): Sort this out |
| 122 | expected = gen_ground_truth(mod, target, dev, inputs) |
| 123 | return mod, inputs, expected |
| 124 | |
| 125 | |
| 126 | entry_func_name = tvm.testing.parameter("main", "func") |
no test coverage detected
searching dependent graphs…