()
| 132 | |
| 133 | |
| 134 | def test_kernel_sharing(): |
| 135 | low, high = -1, 1 |
| 136 | data_np = np.random.randint(low, high, size=(16, 32, 32, 8)).astype("float16") |
| 137 | weight1_np = np.random.randint(low, high, size=(8, 3, 3, 8)).astype("float16") |
| 138 | weight2_np = np.random.randint(low, high, size=(8, 3, 3, 8)).astype("float16") |
| 139 | |
| 140 | out = get_result_with_relax_cutlass_offload( |
| 141 | Conv2dx2, data_np, weight1_np, weight2_np, assert_all_bindings_fused=False |
| 142 | ) |
| 143 | ref = build_and_run(Conv2dx2, [data_np, weight1_np, weight2_np], "llvm") |
| 144 | |
| 145 | np.testing.assert_equal(out, ref) |
| 146 | |
| 147 | |
| 148 | def get_relax_conv2d_module( |
nothing calls this directly
no test coverage detected
searching dependent graphs…