Searched refs:batchCount (Results 1 – 1 of 1) sorted by relevance
| /libCEED/backends/magma/ |
| H A D | ceed-magma-gemm-nontensor.cpp | 41 … magma_int_t lddc, magma_int_t strideC, magma_int_t batchCount, magma_queue_t queue) { in magmablas_gemm_batched_strided() argument
44 (float)beta, (float *)d_C, lddc, strideC, batchCount, queue); in magmablas_gemm_batched_strided()
47 (double)beta, (double *)d_C, lddc, strideC, batchCount, queue); in magmablas_gemm_batched_strided()
69 … magma_int_t strideC, magma_int_t batchCount, magma_queue_t queue) { in devblas_gemm_batched_strided() argument
73 … (const float *)&beta, (float *)d_C, (int)lddc, strideC, (int)batchCount); in devblas_gemm_batched_strided()
77 … (const double *)&beta, (double *)d_C, (int)lddc, strideC, (int)batchCount); in devblas_gemm_batched_strided()
113 magma_int_t batchCount = n / nbatch; in magma_gemm_nontensor() local
114 magma_int_t n2 = n - (batchCount * nbatch); in magma_gemm_nontensor()
120 if (batchCount > 0) { in magma_gemm_nontensor()
122 batchCount, queue); in magma_gemm_nontensor()
[all …]
|