Tests of matmul op where matrices are in batch
()
| 266 | |
| 267 | |
| 268 | def test_batch_matmul(): |
| 269 | """Tests of matmul op where matrices are in batch""" |
| 270 | verify_batch_matmul(16, 16, 235, 128, 1024, cblas) |
| 271 | verify_batch_matmul(16, 16, 235, 128, 1024, cblas, True, False) |
| 272 | verify_batch_matmul(16, 16, 235, 128, 1024, cblas, False, True) |
| 273 | verify_batch_matmul(16, 16, 235, 128, 1024, cblas, True, True) |
| 274 | verify_batch_matmul(16, 16, 235, 128, 1024, mkl) |
| 275 | verify_batch_matmul(16, 16, 235, 128, 1024, mkl, True, False) |
| 276 | verify_batch_matmul(16, 16, 235, 128, 1024, mkl, False, True) |
| 277 | verify_batch_matmul(16, 16, 235, 128, 1024, mkl, True, True) |
| 278 | verify_batch_matmul(16, 1, 235, 128, 1024, cblas) |
| 279 | verify_batch_matmul(1, 16, 235, 128, 1024, cblas) |
| 280 | verify_batch_matmul(16, 1, 235, 128, 1024, cblas) |
| 281 | verify_batch_matmul(1, 16, 235, 128, 1024, cblas) |
| 282 | verify_batch_matmul(16, 1, 235, 128, 1024, mkl) |
| 283 | verify_batch_matmul(1, 16, 235, 128, 1024, mkl) |
| 284 | verify_batch_matmul(16, 1, 235, 128, 1024, mkl) |
| 285 | verify_batch_matmul(1, 16, 235, 128, 1024, mkl) |
| 286 | verify_batch_matmul(1, 1, 1, 16, 3, cblas) |
| 287 | verify_batch_matmul(1, 1, 1, 16, 3, cblas, True, False) |
| 288 | verify_batch_matmul(1, 1, 1, 16, 3, cblas, False, False) |
| 289 | verify_batch_matmul(1, 1, 1, 16, 3, cblas, True, True) |
| 290 | verify_batch_matmul(1, 1, 1, 16, 3, cblas) |
| 291 | verify_batch_matmul(1, 1, 1, 16, 3, mkl) |
| 292 | verify_batch_matmul(1, 1, 1, 16, 3, mkl, True, False) |
| 293 | verify_batch_matmul(1, 1, 1, 16, 3, mkl, False, False) |
| 294 | verify_batch_matmul(1, 1, 1, 16, 3, mkl, True, True) |
| 295 | verify_batch_matmul(1, 1, 1, 16, 3, mkl) |
| 296 | |
| 297 | |
| 298 | if __name__ == "__main__": |
no test coverage detected
searching dependent graphs…