Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #include <stdio.h>
- #include <stdlib.h>
- #include "/opt/intel/mkl/include/mkl.h"
- #include "/opt/intel/mkl/include/mkl_spblas.h"
- #include "/opt/intel/mkl/include/mkl.h"
- int main(int argc, char *argv[])
- {
- int info=1;
- int i;
- // mkl_scsrmultcsr(&trans, &request, &sort, &m, &n, &k, a, ja, ia, b, jb, ib, c, jc, ic, &nzmax, &info);
- int m = 2,nnz = 2;
- float a[] = {2,3};
- int ja[] = {0,1};
- int ia[] = {0,1,2};
- int *ic = (int *)malloc(sizeof(int)*(m+1));
- float *c;
- int *jc;
- char trans = 'N';
- int request = 1;
- int sort = 8;
- int nzmax = 0;
- mkl_scsrmultcsr(&trans, &request, &sort, &m, &m, &m, a, ja, ia, a, ja, ia, c, jc, ic, &nzmax, &info);
- printf("Result nnz = %d\n",ic[m]);
- for(i=0;i<=m;i++)
- {
- printf("ic[%d] = %d\n",i,ic[i]);
- }
- c = (float *)malloc(sizeof(float)*ic[m]);
- jc = (int *)malloc(sizeof(int)*ic[m]);
- request = 2;
- mkl_scsrmultcsr(&trans, &request, &sort, &m, &m, &m, a, ja, ia, a, ja, ia, c, jc, ic, &nzmax, &info);
- for(i=0;i<ic[m];i++)
- {
- printf("%f %d\n",c[i],jc[i]);
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement