mm_to_bcsr.h 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010 Centre National de la Recherche Scientifique
  4. *
  5. * StarPU is free software; you can redistribute it and/or modify
  6. * it under the terms of the GNU Lesser General Public License as published by
  7. * the Free Software Foundation; either version 2.1 of the License, or (at
  8. * your option) any later version.
  9. *
  10. * StarPU is distributed in the hope that it will be useful, but
  11. * WITHOUT ANY WARRANTY; without even the implied warranty of
  12. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
  13. *
  14. * See the GNU Lesser General Public License in COPYING.LGPL for more details.
  15. */
  16. #include <string.h>
  17. #include <stdint.h>
  18. #include <stdio.h>
  19. #include <stdlib.h>
  20. #include "mmio.h"
  21. /* convert a matrix stored in a file with the matrix market format into the
  22. * BCSR format */
  23. typedef struct tmp_block {
  24. /* we have a linked list of blocks */
  25. struct tmp_block *next;
  26. /* column i, row j*/
  27. unsigned i, j;
  28. float *val;
  29. } tmp_block_t;
  30. typedef struct {
  31. unsigned r,c;
  32. unsigned nnz_blocks;
  33. unsigned nrows_blocks;
  34. float *val;
  35. uint32_t *colind;
  36. uint32_t *rowptr;
  37. } bcsr_t;
  38. /* directly read input from a file */
  39. bcsr_t *mm_file_to_bcsr(char *filename, unsigned c, unsigned r);
  40. /* read the matrix as a set of valuated coordinates */
  41. bcsr_t *mm_to_bcsr(unsigned nz, unsigned *I, unsigned *J, float *val, unsigned c, unsigned r);