| H A D | ceed-magma-gemm-nontensor.cpp | 40 … const CeedScalar *d_B, magma_int_t lddb, magma_int_t strideB, CeedScalar beta, CeedScalar *d_C, in magmablas_gemm_batched_strided() argument
43 …ans_B, m, n, k, (float)alpha, (const float *)d_A, ldda, strideA, (const float *)d_B, lddb, strideB, in magmablas_gemm_batched_strided()
46 …_B, m, n, k, (double)alpha, (const double *)d_A, ldda, strideA, (const double *)d_B, lddb, strideB, in magmablas_gemm_batched_strided()
68 … magma_int_t lddb, magma_int_t strideB, CeedScalar beta, CeedScalar *d_C, magma_int_t lddc, in devblas_gemm_batched_strided() argument
72 …nst float *)&alpha, (const float *)d_A, (int)ldda, strideA, (const float *)d_B, (int)lddb, strideB, in devblas_gemm_batched_strided()
76 … double *)&alpha, (const double *)d_A, (int)ldda, strideA, (const double *)d_B, (int)lddb, strideB, in devblas_gemm_batched_strided()
116 magma_int_t strideB = lddb * nbatch; in magma_gemm_nontensor() local
121 …ed(trans_A, trans_B, m, nbatch, k, alpha, d_A, ldda, strideA, d_B, lddb, strideB, beta, d_C, lddc,… in magma_gemm_nontensor()
127 …devblas_gemm(trans_A, trans_B, m, n2, k, alpha, d_A, ldda, d_B + batchCount * strideB, lddb, beta,… in magma_gemm_nontensor()
131 …ed(trans_A, trans_B, m, nbatch, k, alpha, d_A, ldda, strideA, d_B, lddb, strideB, beta, d_C, lddc,… in magma_gemm_nontensor()
[all …]
|