saberbud/ceres-solver
1
0
Fork 0
mirror of https://github.com/ceres-solver/ceres-solver.git synced 2026-08-29 16:40:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

CUDA CGNR, Part 2: CudaVector

Browse Source 
* Added CudaVector to manage and operate on vectors with CUDA.
* Added tests for CudaVector

Change-Id: I528258c8e883011e8709bddf59edb1e933af8060
This commit is contained in:
Joydeep Biswas
2022-08-13 16:20:05 -05:00
parent 3af3dee189
commit c914c7a2b3
6 changed files with 775 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/ceres/context_impl.h
+2 -2
View File
@@ -78,13 +78,13 @@ class CERES_NO_EXPORT ContextImpl final : public Context {
// returned.
bool InitCUDA(std::string* message);
void TearDown();
inline bool IsCUDAInitialized() const { return is_cuda_initialized_; }
cusolverDnHandle_t cusolver_handle_ = nullptr;
cublasHandle_t cublas_handle_ = nullptr;
cudaStream_t stream_ = nullptr;
cusparseHandle_t cusparse_handle_ = nullptr;
// Indicates whether all the CUDA resources have been initialized.
bool cuda_initialized_ = false;
bool is_cuda_initialized_ = false;
#endif // CERES_NO_CUDA
};