cu_orthogonalization.h

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.
 */
 
 
#ifndef _CU_TRACE_ESTIMATOR_CU_ORTHOGONALIZATION_H_
#define _CU_TRACE_ESTIMATOR_CU_ORTHOGONALIZATION_H_
 
// =======
// Imports
// =======
 
#include <cublas_v2.h>  // cublasHandle_t
#include "../_definitions/types.h"  // IndexType, LongIndexType, FlagType
 
 
// ====================
// cu Orthogonalization
// ====================
 
 
template <typename DataType>
class cuOrthogonalization
{
    public:
 
        // Gram-Schmidt Process
        static void gram_schmidt_process(
                cublasHandle_t cublas_handle,
                const DataType* V,
                const LongIndexType vector_size,
                const IndexType num_vectors,
                const IndexType last_vector,
                const FlagType num_ortho,
                DataType* r);
 
        // Orthogonalize Vectors
        static void orthogonalize_vectors(
                cublasHandle_t cublas_handle,
                DataType* vectors,
                const LongIndexType vector_size,
                const IndexType num_vectors);
};
 
#endif  // _CU_TRACE_ESTIMATOR_CU_ORTHOGONALIZATION_H_