| 278 | } |
| 279 | |
| 280 | struct blas_params { |
| 281 | int m, n, k, ld2, ld3, rd2, rd3; |
| 282 | af_dtype type; |
| 283 | blas_params(int m_, int n_, int k_, int ld2_, int ld3_, int rd2_, int rd3_, |
| 284 | af_dtype type_) |
| 285 | : m(m_) |
| 286 | , n(n_) |
| 287 | , k(k_) |
| 288 | , ld2(ld2_) |
| 289 | , ld3(ld3_) |
| 290 | , rd2(rd2_) |
| 291 | , rd3(rd3_) |
| 292 | , type(type_) {} |
| 293 | }; |
| 294 | |
| 295 | class MatrixMultiplyBatch : public ::testing::TestWithParam<blas_params> { |
| 296 | public: |