math.hpp File Reference

#include "data.hpp"
#include "part_descriptor.hpp"
#include <raft/core/comms.hpp>

Include dependency graph for math.hpp:

Go to the source code of this file.

Namespaces
	MLCommon
	MLCommon::Matrix
	MLCommon::Matrix::opg

Functions
template<bool rowMajor, bool bcastAlongRows>
void	MLCommon::Matrix::opg::matrixVectorBinaryDivSkipZero (std::vector< Matrix::Data< double > * > &data, const Matrix::PartDescriptor &inDesc, const Matrix::Data< double > &vec, bool return_zero, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
template<bool rowMajor, bool bcastAlongRows>
void	MLCommon::Matrix::opg::matrixVectorBinaryDivSkipZero (std::vector< Matrix::Data< float > * > &data, const Matrix::PartDescriptor &inDesc, const Matrix::Data< float > &vec, bool return_zero, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
template<bool rowMajor, bool bcastAlongRows>
void	MLCommon::Matrix::opg::matrixVectorBinaryMult (std::vector< Matrix::Data< double > * > &data, const Matrix::PartDescriptor &inDesc, const Matrix::Data< double > &vec, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)
template<bool rowMajor, bool bcastAlongRows>
void	MLCommon::Matrix::opg::matrixVectorBinaryMult (std::vector< Matrix::Data< float > * > &data, const Matrix::PartDescriptor &inDesc, const Matrix::Data< float > &vec, const raft::comms::comms_t &comm, cudaStream_t *streams, int n_streams)