cu_affine_matrix_function.cu

Go to the documentation of this file.

/*
 *  SPDX-FileCopyrightText: Copyright 2021, Siavash Ameli <sameli@berkeley.edu>
 *  SPDX-License-Identifier: BSD-3-Clause
 *  SPDX-FileType: SOURCE
 *
 *  This program is free software: you can redistribute it and/or modify it
 *  under the terms of the license found in the LICENSE.txt file in the root
 *  directory of this source tree.
 */
 
 
// =======
// Headers
// =======
 
#include "./cu_affine_matrix_function.h"
#include <cassert>  // assert
#include "../_definitions/debugging.h"  // ASSERT
#include "../_cu_basic_algebra/cu_vector_operations.h"  // cuVectorOperations
#include "../_cuda_utilities/cuda_interface.h"  // CudaInterface
 
 
// ===========
// constructor
// ===========
 
 
template <typename DataType>
cuAffineMatrixFunction<DataType>::cuAffineMatrixFunction():
    B_is_identity(false)
{
    // This class has one parameter that is t in A+tB
    this->num_parameters = 1;
}
 
 
// ==========
// destructor
// ==========
 
 
template <typename DataType>
cuAffineMatrixFunction<DataType>::~cuAffineMatrixFunction()
{
}
 
 
// ==============
// get eigenvalue
// ==============
 
 
template <typename DataType>
DataType cuAffineMatrixFunction<DataType>::get_eigenvalue(
        const DataType* known_parameters,
        const DataType known_eigenvalue,
        const DataType* inquiry_parameters) const
{
    ASSERT((this->eigenvalue_relation_known == 1),
            "An eigenvalue relation is not known. This function should be "
            "called only when the matrix B is a scalar multiple of the "
            "identity matrix");
 
    // Shift the eigenvalue by the parameter
    DataType inquiry_eigenvalue = \
        known_eigenvalue - known_parameters[0] + inquiry_parameters[0];
 
    return inquiry_eigenvalue;
}
 
 
// =================
// add scaled vector
// =================
 
 
template <typename DataType>
void cuAffineMatrixFunction<DataType>::_add_scaled_vector(
        const DataType* input_vector,
        const LongIndexType vector_size,
        const DataType scale,
        DataType* output_vector) const
{
    // Get device id
    int device_id = CudaInterface<DataType>::get_device();
 
    // Subtracting two vectors with minus scale sign, which is adding.
    cuVectorOperations<DataType>::subtract_scaled_vector(
            this->cublas_handle[device_id], input_vector, vector_size, -scale,
            output_vector);
}
 
 
// ===============================
// Explicit template instantiation
// ===============================
 
template class cuAffineMatrixFunction<float>;
template class cuAffineMatrixFunction<double>;