mm_to_bcsr.h 1.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354
  1. /* StarPU --- Runtime system for heterogeneous multicore architectures.
  2. *
  3. * Copyright (C) 2010, 2011 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. {
  25. /* we have a linked list of blocks */
  26. struct tmp_block *next;
  27. /* column i, row j*/
  28. unsigned i, j;
  29. float *val;
  30. } tmp_block_t;
  31. typedef struct
  32. {
  33. unsigned r,c;
  34. unsigned nnz_blocks;
  35. unsigned nrows_blocks;
  36. float *val;
  37. uint32_t *colind;
  38. uint32_t *rowptr;
  39. } bcsr_t;
  40. /* directly read input from a file */
  41. bcsr_t *mm_file_to_bcsr(char *filename, unsigned c, unsigned r);
  42. /* read the matrix as a set of valuated coordinates */
  43. bcsr_t *mm_to_bcsr(unsigned nz, unsigned *I, unsigned *J, float *val, unsigned c, unsigned r);