You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
111 lines
3.9 KiB
C
111 lines
3.9 KiB
C
#pragma once
|
|
|
|
/*
|
|
* spGPU - Sparse matrices on GPU library.
|
|
*
|
|
* Copyright (C) 2010 - 2012
|
|
* Davide Barbieri - University of Rome Tor Vergata
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* version 3 as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*/
|
|
|
|
#include "ell.h"
|
|
#include <string.h>
|
|
|
|
/** \addtogroup conversionRoutines Conversion Routines
|
|
* @{
|
|
*/
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/**
|
|
* \fn void computeEllRowLenghts(int *ellRowLengths, int *ellMaxRowSize, int rowsCount, int nonZerosCount, const int* cooRowIndices, int cooBaseIndex)
|
|
* Compute the Ell row lengths array (and the greatest row size) from the COO matrix format.
|
|
* \param ellRowLengths Array of length rowsCount to be filled by the non zeros count for every matrix row
|
|
* \param ellMaxRowSize outputs the greatest row size (in non zeros)
|
|
* \param rowsCount the number of rows of the coo matrix to convert
|
|
* \param nonZerosCount the non zeros count of the coo matrix to convert
|
|
* \param cooRowIndices the row indices array for the coo matrix to convert
|
|
* \param cooBaseIndex the input base index (e.g. 0 for C, 1 for Fortran)
|
|
*/
|
|
void computeEllRowLenghts(
|
|
int *ellRowLengths,
|
|
int *ellMaxRowSize,
|
|
int rowsCount,
|
|
int nonZerosCount,
|
|
const int* cooRowIndices,
|
|
int cooBaseIndex
|
|
);
|
|
|
|
/**
|
|
* \fn int computeEllAllocPitch(int rowsCount)
|
|
* This function returns a pitch (in number of elements) that can be used to allocate both indices and values arrays for ELL matrix format.
|
|
* \param rowsCount the rows count
|
|
* \return the pitch for an ELL matrix of rowsCount rows.
|
|
*/
|
|
int computeEllAllocPitch(int rowsCount);
|
|
|
|
|
|
/**
|
|
* \fn void cooToEll(void *ellValues,int *ellIndices,int ellValuesPitch,int ellIndicesPitch,int ellMaxRowSize,int ellBaseIndex,int rowsCount,int nonZerosCount,const int* cooRowIndices,const int* cooColsIndices,const void* cooValues,int cooBaseIndex, spgpuType_t valuesType)
|
|
* Convert a matrix in COO format to a matrix in ELL format.
|
|
* The matrix is stored in column-major format. The ellValues and ellIndices sizes are ellMaxRowSize * pitch (pitch is in bytes).
|
|
* \param ellValues pointer to the area that will be filled by the non zero coefficients
|
|
* \param ellIndices pointer to the area that will be filled by the non zero indices
|
|
* \param ellValuesPitch the column-major allocation's pitch of ellValues (in number of elements)
|
|
* \param ellIndicesPitch the column-major allocation's pitch of ellIndices (in number of elements)
|
|
* \param ellMaxRowSize the greatest row size
|
|
* \param ellBaseIndex the desired base index for the ELL matrix (e.g. 0 for C, 1 for Fortran)
|
|
* \param rowsCount input matrix rows count
|
|
* \param nonZerosCount input matrix non zeros count
|
|
* \param cooRowIndices input matrix row indices pointer
|
|
* \param cooColsIndices input matrix column indices pointer
|
|
* \param cooValues input matrix non zeros values pointer
|
|
* \param cooBaseIndex input matrix base index
|
|
* \param valuesType the type for elements in ellValues and cooValues (i.e. SPGPU_TYPE_FLOAT or SPGPU_TYPE_DOUBLE)
|
|
*/
|
|
void cooToEll(
|
|
void *ellValues,
|
|
int *ellIndices,
|
|
int ellValuesPitch,
|
|
int ellIndicesPitch,
|
|
int ellMaxRowSize,
|
|
int ellBaseIndex,
|
|
int rowsCount,
|
|
int nonZerosCount,
|
|
const int* cooRowIndices,
|
|
const int* cooColsIndices,
|
|
const void* cooValues,
|
|
int cooBaseIndex,
|
|
spgpuType_t valuesType
|
|
);
|
|
|
|
void ellToOell(
|
|
int *rIdx,
|
|
void *dstEllValues,
|
|
int *dstEllIndices,
|
|
int *dstRs,
|
|
const void *srcEllValues,
|
|
const int *srcEllIndices,
|
|
const int *srcRs,
|
|
int ellValuesPitch,
|
|
int ellIndicesPitch,
|
|
int rowsCount,
|
|
spgpuType_t valuesType
|
|
);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
/** @}*/
|