- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Hi all,
I am trying to use mkl_sparse_spmm in a function, to calculate C = A*B, where A, B, C are in CSR format.
I would like to return the information of C, i.e. rowIndex_C, column_C and values_C through the pointers that I passed to the function arguments.
However, it looks like only within the function, the info of C exists. When I try to print values_C outside of the function, the pointer seems to be NULL. It looks like that the memory is allocated by mkl_sparse_d_export_csr within the function, but no longer exists after the function call.
Here is the function:
void spmm(int M, int N, int K, int* rowIndex_A, int* columns_A, double* values_A, int* rowIndex_B, int* columns_B, double* values_B, int* pointerB_C, int* pointerE_C, int* columns_C, double* values_C){ int rows, cols; sparse_index_base_t indexing; sparse_matrix_t csrA = NULL, csrB = NULL, csrC = NULL; mkl_sparse_d_create_csr( &csrA, SPARSE_INDEX_BASE_ZERO, M, K, rowIndex_A, rowIndex_A+1, columns_A, values_A ); mkl_sparse_d_create_csr( &csrB, SPARSE_INDEX_BASE_ZERO, K, N, rowIndex_B, rowIndex_B+1, columns_B, values_B ); mkl_sparse_spmm( SPARSE_OPERATION_NON_TRANSPOSE, csrA, csrB, &csrC ); mkl_sparse_d_export_csr( csrC, &indexing, &rows, &cols, &pointerB_C, &pointerE_C, &columns_C, &values_C ) mkl_sparse_destroy( csrA ); mkl_sparse_destroy( csrB ); mkl_sparse_destroy( csrC ); }
Any suggestions on solving this issue? Thanks a lot! A relevant question is how to use spmm in python/scipy, which is purpose why I wrote such a function.
Best,
Zhihao
Link Copied
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Hi,
The memory for csrC and for the CSR representation, that you want to return (i.e. pointerB_C, pointerE_C, columns_C, values_C) is indeed allocated inside the mkl_sparse_spmm() routine.
In order to use the data outside of spmm() function, you should go with the next interface, using double pointers for C data:
void spmm(int M, int N, int K, int* rowIndex_A, int* columns_A, double* values_A, int* rowIndex_B, int* columns_B, double* values_B, int** pointerB_C, int** pointerE_C, int** columns_C, double** values_C)
Hope, this will help!
Best regards,
Maria
- Mark as New
- Bookmark
- Subscribe
- Mute
- Subscribe to RSS Feed
- Permalink
- Report Inappropriate Content
Zhukova, Maria (Intel) wrote:
Hi,
The memory for csrC and for the CSR representation, that you want to return (i.e. pointerB_C, pointerE_C, columns_C, values_C) is indeed allocated inside the mkl_sparse_spmm() routine.
In order to use the data outside of spmm() function, you should go with the next interface, using double pointers for C data:
void spmm(int M, int N, int K, int* rowIndex_A, int* columns_A, double* values_A, int* rowIndex_B, int* columns_B, double* values_B, int** pointerB_C, int** pointerE_C, int** columns_C, double** values_C)Hope, this will help!
Best regards,
Maria
Thanks for your reply, it works. The only question in Python side is that you have to free the memory by hand in this case. Is there already some API for MKL Sparse Blas in Python?
Best,
Zhihao

- Subscribe to RSS Feed
- Mark Topic as New
- Mark Topic as Read
- Float this Topic for Current User
- Bookmark
- Subscribe
- Printer Friendly Page