Tests of matmul+add op
()
| 95 | |
| 96 | |
| 97 | def test_matmul_add(): |
| 98 | """Tests of matmul+add op""" |
| 99 | verify_matmul_add(235, 128, 1024, cblas) |
| 100 | verify_matmul_add(235, 128, 1024, cblas, True, False) |
| 101 | verify_matmul_add(235, 128, 1024, cblas, False, True) |
| 102 | verify_matmul_add(235, 128, 1024, cblas, True, True) |
| 103 | verify_matmul_add(235, 128, 1024, mkl) |
| 104 | verify_matmul_add(235, 128, 1024, mkl, True, False) |
| 105 | verify_matmul_add(235, 128, 1024, mkl, False, True) |
| 106 | verify_matmul_add(235, 128, 1024, mkl, True, True) |
| 107 | verify_matmul_add(235, 128, 1024, dnnl) |
| 108 | verify_matmul_add(235, 128, 1024, dnnl, True, False) |
| 109 | verify_matmul_add(235, 128, 1024, dnnl, False, True) |
| 110 | verify_matmul_add(235, 128, 1024, dnnl, True, True) |
| 111 | verify_matmul_add(1, 16, 4, cblas) |
| 112 | verify_matmul_add(1, 16, 3, cblas, True, False) |
| 113 | verify_matmul_add(1, 16, 3, cblas, False, False) |
| 114 | verify_matmul_add(1, 16, 3, cblas, True, True) |
| 115 | verify_matmul_add(1, 16, 4, mkl) |
| 116 | verify_matmul_add(1, 16, 3, mkl, True, False) |
| 117 | verify_matmul_add(1, 16, 3, mkl, False, False) |
| 118 | verify_matmul_add(1, 16, 3, mkl, True, True) |
| 119 | verify_matmul_add(1, 16, 4, dnnl) |
| 120 | verify_matmul_add(1, 16, 3, dnnl, True, False) |
| 121 | verify_matmul_add(1, 16, 3, dnnl, False, False) |
| 122 | verify_matmul_add(1, 16, 3, dnnl, True, True) |
| 123 | |
| 124 | |
| 125 | def verify_quantized_matmul_add(matrix_m, matrix_l, matrix_n, transa=False, transb=False): |
no test coverage detected
searching dependent graphs…