Lines Matching refs:queue
26 magma_int_t lddc, magma_queue_t queue) { in magmablas_gemm() argument
29 queue); in magmablas_gemm()
32 queue); in magmablas_gemm()
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()
55 magma_int_t lddc, magma_queue_t queue) { in devblas_gemm() argument
57 …alpha, (const float *)d_A, ldda, (const float *)d_B, lddb, (float)beta, (float *)d_C, lddc, queue); in devblas_gemm()
60 queue); in devblas_gemm()
69 … magma_int_t strideC, magma_int_t batchCount, magma_queue_t queue) { in devblas_gemm_batched_strided() argument
71 …devblasSgemmStridedBatched(magma_queue_get_devblas_handle(queue), devblas_trans_const(trans_A), de… in devblas_gemm_batched_strided()
75 …devblasDgemmStridedBatched(magma_queue_get_devblas_handle(queue), devblas_trans_const(trans_A), de… in devblas_gemm_batched_strided()
85 magma_int_t lddc, magma_queue_t queue) { in magma_gemm_nontensor() argument
94 devblas_gemm(trans_A, trans_B, m, n, k, alpha, d_A, ldda, d_B, lddb, beta, d_C, lddc, queue); in magma_gemm_nontensor()
107 … magmablas_gemm(trans_A, trans_B, m, n, k, alpha, d_A, ldda, d_B, lddb, beta, d_C, lddc, queue); in magma_gemm_nontensor()
109 devblas_gemm(trans_A, trans_B, m, n, k, alpha, d_A, ldda, d_B, lddb, beta, d_C, lddc, queue); in magma_gemm_nontensor()
122 batchCount, queue); in magma_gemm_nontensor()
127 …alpha, d_A, ldda, d_B + batchCount * strideB, lddb, beta, d_C + batchCount * strideC, lddc, queue); in magma_gemm_nontensor()
132 batchCount, queue); in magma_gemm_nontensor()
138 d_C + batchCount * strideC, lddc, strideC, 1, queue); in magma_gemm_nontensor()
144 ceed_magma_queue_sync(queue); in magma_gemm_nontensor()