()
| 102 | |
| 103 | |
| 104 | def test_output(): |
| 105 | p = migraphx.parse_onnx("conv_relu_maxpool_test.onnx") |
| 106 | p.compile(migraphx.get_target("gpu")) |
| 107 | |
| 108 | r1 = run(p)[-1] |
| 109 | r2 = run(p)[-1] |
| 110 | assert_eq(r1, r2) |
| 111 | assert_eq(r1.tolist(), r2.tolist()) |
| 112 | |
| 113 | check_argument(r1) |
| 114 | check_argument(r2) |
| 115 | |
| 116 | m1 = memoryview(r1) |
| 117 | m2 = memoryview(r2) |
| 118 | |
| 119 | check_shapes(r1, m1) |
| 120 | check_shapes(r2, m2) |
| 121 | |
| 122 | |
| 123 | test_input() |
no test coverage detected