MCPcopy Create free account
hub / github.com/NVIDIA/cuda-samples / cublasXgemmBatched

Function cublasXgemmBatched

cpp/4_CUDA_Libraries/batchCUBLAS/batchCUBLAS.cpp:230–252  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

228}
229
230static 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
254static inline cublasStatus_t cublasXgemmBatched(cublasHandle_t handle,
255 cublasOperation_t transa,

Callers 1

test_gemm_loopFunction · 0.85

Calls

no outgoing calls

Tested by 1

test_gemm_loopFunction · 0.68