Lines Matching refs:batchCount
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()
127 …rans_A, trans_B, m, n2, k, alpha, d_A, ldda, d_B + batchCount * strideB, lddb, beta, d_C + batchCo… in magma_gemm_nontensor()
130 if (batchCount > 0) { in magma_gemm_nontensor()
132 batchCount, queue); in magma_gemm_nontensor()
137 …d_strided(trans_A, trans_B, m, n2, k, alpha, d_A, ldda, strideA, d_B + batchCount * strideB, lddb,… in magma_gemm_nontensor()
138 d_C + batchCount * strideC, lddc, strideC, 1, queue); in magma_gemm_nontensor()