vector_operations.hpp
Go to the documentation of this file.
58T innerProductAtIndices(cublasHandle_t cublasHandle, const T* deviceA, const T* deviceB, T* buffer, size_t numberOfElements, const int* indices);
Definition: autotuner.hpp:29 void syncFromRecvBuf(T *deviceA, T *buffer, size_t numberOfElements, const int *indices) T innerProductAtIndices(cublasHandle_t cublasHandle, const T *deviceA, const T *deviceB, T *buffer, size_t numberOfElements, const int *indices) innerProductAtIndices computes the inner product between deviceA[indices] and deviceB[indices] void setZeroAtIndexSet(T *deviceData, size_t numberOfElements, const int *indices) setZeroAtIndexSet sets deviceData to zero in the indices of contained in indices void prepareSendBuf(const T *deviceA, T *buffer, size_t numberOfElements, const int *indices) void setVectorValue(T *deviceData, size_t numberOfElements, const T &value) setVectorValue sets every element of deviceData to value void weightedDiagMV(const T *squareBlockVector, const size_t numberOfRows, const size_t blocksize, T relaxationFactor, const T *srcVec, T *dstVec) Compue the weighted matrix vector product where the matrix is diagonal, the diagonal is a vector,... |