cuda_solvers.cc | 366 int lwork; local 368 bufsize(cusolver_dn_handle, uplo, n, CUDAComplex(A), lda, &lwork)); 371 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); 375 CUDAComplex(dev_workspace.mutable_data()), lwork, dev_lapack_info)); 399 int lwork; local 401 bufsize(cusolver_dn_handle, m, n, CUDAComplex(A), lda, &lwork)); 404 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false); 459 int lwork; local 461 bufsize(cusolver_dn_handle, m, n, CUDAComplex(A), lda, &lwork)); 464 cuda_solver->GetScratchSpace<Scalar>(lwork, "", /* on_host */ false) 494 int lwork; local 538 int lwork; local 579 int lwork; local 619 int lwork; local 657 int lwork; local [all...] |