cu_trace_estimator.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_TRACE_ESTIMATOR_H_
#define _CU_TRACE_ESTIMATOR_CU_TRACE_ESTIMATOR_H_
 
// ======
// Headers
// ======
 
#include "../functions/functions.h"  // Function
#include "../_cu_linear_operator/cu_linear_operator.h"  // cuLinearOperator
#include "../_definitions/types.h"  // IndexType, FlagType
#include "../_random_generator/random_number_generator.h"  // RandomNumberGe...
 
 
// =================
// c Trace Estimator
// =================
 
 
template <typename DataType>
class cuTraceEstimator
{
    public:
 
        // c trace estimator
        static FlagType cu_trace_estimator(
                cuLinearOperator<DataType>* A,
                DataType* parameters,
                const IndexType num_inquiries,
                const Function* matrix_function,
                const FlagType gram,
                const DataType exponent,
                const FlagType orthogonalize,
                const int64_t seed,
                const IndexType lanczos_degree,
                const DataType lanczos_tol,
                const IndexType min_num_samples,
                const IndexType max_num_samples,
                const DataType error_atol,
                const DataType error_rtol,
                const DataType confidence_level,
                const DataType outlier_significance_level,
                const IndexType num_threads,
                const IndexType num_gpu_devices,
                DataType* trace,
                DataType* error,
                DataType** samples,
                IndexType* processed_samples_indices,
                IndexType* num_samples_used,
                IndexType* num_outliers,
                FlagType* converged,
                float& alg_wall_time);
 
    private:
 
        // _c stochastic lanczos quadrature
        static void _cu_stochastic_lanczos_quadrature(
                cuLinearOperator<DataType>* A,
                DataType* parameters,
                const IndexType num_inquiries,
                const Function* matrix_function,
                const FlagType gram,
                const DataType exponent,
                const FlagType orthogonalize,
                const IndexType lanczos_degree,
                const DataType lanczos_tol,
                RandomNumberGenerator& random_number_generator,
                DataType* random_vector,
                FlagType* converged,
                DataType* trace_estimate);
};
 
#endif  // _CU_TRACE_ESTIMATOR_CU_TRACE_ESTIMATOR_H_