LinalgUtils.cpp

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#include "cloudViewer/core/linalg/LinalgUtils.h"
 
#include "cloudViewer/core/CUDAUtils.h"
 
namespace cloudViewer {
namespace core {
 
CuSolverContext& CuSolverContext::GetInstance() {
    static CuSolverContext instance;
    return instance;
}
 
CuSolverContext::CuSolverContext() {
    for (const Device& device : Device::GetAvailableCUDADevices()) {
        CUDAScopedDevice scoped_device(device);
        cusolverDnHandle_t handle;
        if (cusolverDnCreate(&handle) != CUSOLVER_STATUS_SUCCESS) {
            utility::LogError("Unable to create cuSolver handle for {}.",
                              device.ToString());
        }
        map_device_to_handle_[device] = handle;
    }
}
 
CuSolverContext::~CuSolverContext() {
    // Destroy map_device_to_handle_
    for (auto& item : map_device_to_handle_) {
        if (cusolverDnDestroy(item.second) != CUSOLVER_STATUS_SUCCESS) {
            utility::LogError(
                    "Unable to destroy cuSolver handle for device {}.",
                    item.first.ToString());
        }
    }
}
 
cusolverDnHandle_t& CuSolverContext::GetHandle(const Device& device) {
    if (device.GetType() != Device::DeviceType::CUDA) {
        utility::LogError("cuSolver is only available on CUDA devices");
    }
    if (map_device_to_handle_.count(device) == 0) {
        utility::LogError("cuSolver handle not found for device: {}",
                          device.ToString());
    }
    return map_device_to_handle_.at(device);
}
 
CuBLASContext& CuBLASContext::GetInstance() {
    static CuBLASContext instance;
    return instance;
}
 
CuBLASContext::CuBLASContext() {
    for (const Device& device : Device::GetAvailableCUDADevices()) {
        CUDAScopedDevice scoped_device(device);
        cublasHandle_t handle;
        if (cublasCreate(&handle) != CUBLAS_STATUS_SUCCESS) {
            utility::LogError("Unable to create cublas handle for {}.",
                              device.ToString());
        }
        map_device_to_handle_[device] = handle;
    }
}
 
CuBLASContext::~CuBLASContext() {
    // Destroy map_device_to_handle_
    for (auto& item : map_device_to_handle_) {
        if (cublasDestroy(item.second) != CUBLAS_STATUS_SUCCESS) {
            utility::LogError("Unable to destroy cublas handle for device {}.",
                              item.first.ToString());
        }
    }
}
 
cublasHandle_t& CuBLASContext::GetHandle(const Device& device) {
    if (device.GetType() != Device::DeviceType::CUDA) {
        utility::LogError("cuBLAS is only available on CUDA devices");
    }
    if (map_device_to_handle_.count(device) == 0) {
        utility::LogError("cuBLAS handle not found for device: {}",
                          device.ToString());
    }
    return map_device_to_handle_.at(device);
}
 
}  // namespace core
}  // namespace cloudViewer