dense.hpp File Reference

#include "base.hpp"
#include <raft/core/handle.hpp>
#include <raft/linalg/add.cuh>
#include <raft/linalg/ternary_op.cuh>
#include <raft/util/cuda_utils.cuh>
#include <raft/util/cudart_utils.hpp>
#include <iostream>
#include <vector>
#include <raft/linalg/detail/cublas_wrappers.hpp>
#include <raft/linalg/map_then_reduce.cuh>
#include <raft/linalg/norm.cuh>
#include <raft/linalg/unary_op.cuh>
#include <rmm/device_uvector.hpp>

Include dependency graph for dense.hpp:

Go to the source code of this file.

Classes
struct	ML::SimpleDenseMat< T >
struct	ML::SimpleVec< T >
struct	ML::SimpleVecOwning< T >
struct	ML::SimpleMatOwning< T >

Namespaces
	ML

Enumerations
enum	ML::STORAGE_ORDER { ML::COL_MAJOR = 0 , ML::ROW_MAJOR = 1 }

Functions
template<typename T >
void	ML::col_ref (const SimpleDenseMat< T > &mat, SimpleVec< T > &mask_vec, int c)
template<typename T >
void	ML::col_slice (const SimpleDenseMat< T > &mat, SimpleDenseMat< T > &mask_mat, int c_from, int c_to)
template<typename T >
T	ML::dot (const SimpleVec< T > &u, const SimpleVec< T > &v, T *tmp_dev, cudaStream_t stream)
template<typename T >
T	ML::squaredNorm (const SimpleVec< T > &u, T *tmp_dev, cudaStream_t stream)
template<typename T >
T	ML::nrmMax (const SimpleVec< T > &u, T *tmp_dev, cudaStream_t stream)
template<typename T >
T	ML::nrm2 (const SimpleVec< T > &u, T *tmp_dev, cudaStream_t stream)
template<typename T >
T	ML::nrm1 (const SimpleVec< T > &u, T *tmp_dev, cudaStream_t stream)
template<typename T >
std::ostream &	ML::operator<< (std::ostream &os, const SimpleVec< T > &v)
template<typename T >
std::ostream &	ML::operator<< (std::ostream &os, const SimpleDenseMat< T > &mat)