Searched refs:nbatch (Results 1 – 1 of 1) sorted by relevance
| /libCEED/backends/magma/ |
| H A D | ceed-magma-gemm-nontensor.cpp | 86 magma_int_t nbatch, use_magmablas; in magma_gemm_nontensor() local
101 gemm_selector(arch, precision, trans, m, n, k, &nbatch, &use_magmablas); in magma_gemm_nontensor()
104 if (nbatch == n) { in magma_gemm_nontensor()
113 magma_int_t batchCount = n / nbatch; in magma_gemm_nontensor()
114 magma_int_t n2 = n - (batchCount * nbatch); in magma_gemm_nontensor()
116 magma_int_t strideB = lddb * nbatch; in magma_gemm_nontensor()
117 magma_int_t strideC = lddc * nbatch; in magma_gemm_nontensor()
121 …magmablas_gemm_batched_strided(trans_A, trans_B, m, nbatch, k, alpha, d_A, ldda, strideA, d_B, ldd… in magma_gemm_nontensor()
131 …devblas_gemm_batched_strided(trans_A, trans_B, m, nbatch, k, alpha, d_A, ldda, strideA, d_B, lddb,… in magma_gemm_nontensor()
|