| 228 | } |
| 229 | |
| 230 | static inline cublasStatus_t cublasXgemmBatched(cublasHandle_t handle, |
| 231 | cublasOperation_t transa, |
| 232 | cublasOperation_t transb, |
| 233 | int m, |
| 234 | int n, |
| 235 | int k, |
| 236 | float *alpha, |
| 237 | const float *Aarray[], |
| 238 | int lda, |
| 239 | const float *Barray[], |
| 240 | int ldb, |
| 241 | float *beta, |
| 242 | float *Carray[], |
| 243 | int ldc, |
| 244 | int batchCount) |
| 245 | { |
| 246 | #if CUDART_VERSION >= 4010 |
| 247 | return cublasSgemmBatched( |
| 248 | handle, transa, transb, m, n, k, alpha, Aarray, lda, Barray, ldb, beta, Carray, ldc, batchCount); |
| 249 | #else |
| 250 | return CUBLAS_STATUS_SUCCESS; |
| 251 | #endif |
| 252 | } |
| 253 | |
| 254 | static inline cublasStatus_t cublasXgemmBatched(cublasHandle_t handle, |
| 255 | cublasOperation_t transa, |
no outgoing calls