Lines Matching refs:lda
207 auto &lda = MatIMPLCast(m)->lda; in SetPreallocation_() local
216 if (lda <= 0) lda = nrows; in SetPreallocation_()
225 size = lda * ncols; in SetPreallocation_()
254 const auto lda = mimpl->lda; in HostToDevice_() local
258 if (lda > nrows) { in HostToDevice_()
259 …PetscCall(PetscCUPMMemcpy2DAsync(dest, lda, src, lda, nrows, ncols, cupmMemcpyHostToDevice, stream… in HostToDevice_()
261 PetscCall(PetscCUPMMemcpyAsync(dest, src, lda * ncols, cupmMemcpyHostToDevice, stream)); in HostToDevice_()
290 const auto lda = mimpl->lda; in DeviceToHost_() local
294 if (lda > nrows) { in DeviceToHost_()
295 …PetscCall(PetscCUPMMemcpy2DAsync(dest, lda, src, lda, nrows, ncols, cupmMemcpyDeviceToHost, stream… in DeviceToHost_()
297 PetscCall(PetscCUPMMemcpyAsync(dest, src, lda * ncols, cupmMemcpyDeviceToHost, stream)); in DeviceToHost_()
388 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in Factor() local
396 return cupmSolverXgetrf_bufferSize(handle, m, n, da.cupmdata(), lda, fact_lwork); in Factor()
404 …PetscCallCUPMSOLVER(cupmSolverXgetrf(handle, m, n, da.cupmdata(), lda, mcu->d_fact_work, mcu->d_fa… in Factor()
427 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in Solve() local
436 handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, lwork in Solve()
442 …PetscCallCUPMSOLVER(cupmSolverXgetrs(handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, m… in Solve()
473 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in Factor() local
482 handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, fact_lwork in Factor()
489 …VER(cupmSolverXpotrf(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, mcu->d_fact_work, … in Factor()
501 …PetscCallCUPMSOLVER(cupmSolverDnXsytrf_bufferSize(handle, n, da.cupmdata(), lda, &mcu->d_fact_lwor… in Factor()
506 …PetscCallCUPMSOLVER(cupmSolverXsytrf(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da, lda, mcu->d_fact_i… in Factor()
526 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in Solve() local
535 handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), lda, x, ldx, lwork in Solve()
541 …pmSolverXpotrs(handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), lda, x, ldx, mcu->d_fac… in Solve()
575 const auto lda = static_cast<cupmBlasInt_t>(mimpl->lda); in Factor() local
585 return cupmSolverXgeqrf_bufferSize(handle, m, n, da.cupmdata(), lda, fact_lwork); in Factor()
591 …PetscCallCUPMSOLVER(cupmSolverXgeqrf(handle, m, n, da.cupmdata(), lda, mcu->d_fact_tau, mcu->d_fac… in Factor()
619 const auto lda = static_cast<cupmBlasInt_t>(mimpl->lda); in Solve() local
622 …_MODE_UPPER, CUPMBLAS_OP_T, CUPMBLAS_DIAG_NON_UNIT, rank, nrhs, &one, da.cupmdata(), lda, x, ldx)); in Solve()
623 …ndle, CUPMSOLVER_SIDE_LEFT, CUPMSOLVER_OP_N, m, nrhs, rank, da.cupmdata(), lda, fact_tau, x, ldx, … in Solve()
628 …mqr(solver_handle, CUPMSOLVER_SIDE_LEFT, op, m, nrhs, rank, da.cupmdata(), lda, fact_tau, x, ldx, … in Solve()
630 …_MODE_UPPER, CUPMBLAS_OP_N, CUPMBLAS_DIAG_NON_UNIT, rank, nrhs, &one, da.cupmdata(), lda, x, ldx)); in Solve()
809 const auto lda = static_cast<cupmBlasInt_t>(MatIMPLCast(A)->lda); in MatMultAddColumnRange_Dispatch_() local
862 …PetscCallCUPMBLAS(cupmBlasXgemv(handle, op, m, n, &one, da.cupmdata() + c_start * lda, lda, dxx.cu… in MatMultAddColumnRange_Dispatch_()
1359 const auto lda_a = MatIMPLCast(A)->lda; in Copy()
1360 const auto lda_b = MatIMPLCast(B)->lda; in Copy()
1388 const auto lda = MatIMPLCast(m)->lda; in ZeroEntries() local
1392 if (lda > ma) { in ZeroEntries()
1393 PetscCall(PetscCUPMMemset2DAsync(va.data(), lda, 0, ma, na, stream)); in ZeroEntries()
1414 …PETSC_HOSTDEVICE_INLINE_DECL T operator()(T x) const noexcept { return ((x / nrows) * lda) + (x % … in operator ()()
1418 PetscInt lda; member
1427 …rixIterator(Iterator first, Iterator last, PetscInt nrows, PetscInt ncols, PetscInt lda) noexcept : in SubMatrixIterator()
1429 std::move(first), std::move(last), {nrows, ncols, lda} in SubMatrixIterator()
1441 …ator(PetscInt rstart, PetscInt rend, PetscInt cstart, PetscInt cend, PetscInt lda, T *ptr) noexcept in make_submat_iterator() argument
1447 …return {dptr + (rstart * lda) + cstart, dptr + ((rstart + nrows) * lda) + cstart, nrows, ncols, ld… in make_submat_iterator()
1473 const auto lda = MatIMPLCast(A)->lda; in Conjugate() local
1477 if (lda > m) { in Conjugate()
1480 const auto sub_mat = detail::make_submat_iterator(0, m, 0, n, lda, da.data()); in Conjugate()
1523 const auto lda = MatIMPLCast(A)->lda; in Scale() local
1525 if (lda > m) { in Scale()
1531 const auto sub_mat = detail::make_submat_iterator(0, m, 0, n, lda, da.data()); in Scale()
1570 const auto lda_x = MatIMPLCast(X)->lda; in AXPY()
1571 const auto lda_y = MatIMPLCast(Y)->lda; in AXPY()
1639 PetscInt lda; in SetRandom() local
1641 PetscCall(MatDenseGetLDA(A, &lda)); in SetRandom()
1642 if (lda > m) { in SetRandom()
1643 … for (PetscInt i = 0; i < n; i++) PetscCall(PetscRandomGetValues(rng, m, a.data() + i * lda)); in SetRandom()
1664 …auto col_offset = [&](const PetscScalar *ptr) { return ptr + col * MatIMPLCast(A)->lda; }; in GetColumnVector()
1711 …pl->cvec, mimpl->ptrinuse + static_cast<std::size_t>(col) * static_cast<std::size_t>(mimpl->lda))); in GetColumnVec()
1803 const auto lda = static_cast<cupmBlasInt_t>(mimpl->lda); in InvertFactors() local
1806 …OLVER(cupmSolverXpotri_bufferSize(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, &il)); in InvertFactors()
1813 …PetscCallCUPMSOLVER(cupmSolverXpotri(handle, CUPMSOLVER_FILL_MODE_LOWER, n, da.cupmdata(), lda, mc… in InvertFactors()
1836 …t = [&](PetscScalar *ptr) { return ptr + rbegin + static_cast<std::size_t>(cbegin) * mimpl->lda; }; in GetSubMatrix()
1860 PetscCall(MatDenseSetLDA(cmat, mimpl->lda)); in GetSubMatrix()