英特尔MKL SpareBlas mm基于CSR的单一索引无法正常工作

时间:2014-10-23 20:26:04

标签: c blas csr intel-mkl

我正在C测试程序中测试英特尔MKL的功能,我发现我无法使备用 mkl_scsrmm 功能CSR < strong> 基于单一 的索引工作。我正在使用CSR与val,columns,pntrb和pntre变体。原始示例放在:

  

&#34; ... MKL \例子\ examples_core_c \ spblasc \源\ cspblas_scsr.c&#34;

这是基于零的索引的第一个代码:

示例#1

#include <stdio.h>
#include "mkl_types.h"
#include "mkl_spblas.h"

int main() {
    #define M 2        
    #define NNZ 4

        MKL_INT        m = M, nnz = NNZ;

        float        values[NNZ]     = {2.0,4.0,4.0,2.0};
        MKL_INT        columns[NNZ]  = {1,2,1,2};
        MKL_INT        rowIndex[M+1] = {1,3,5};

    #define N 2      

        MKL_INT        n = N;
        float         b[M][N]    = {2.0, 1.0, 5.0, 2.0};
        float         c[M][N]    = {0.0, 0.0, 0.0, 0.0};
        float         alpha = 1.0, beta = 0.0;
        char          transa, uplo, nonunit;
        char          matdescra[6];
        MKL_INT       i, j, is;

    transa = 'N';

        matdescra[0] = 'S';
        matdescra[1] = 'L';
        matdescra[2] = 'N';
        matdescra[3] = 'F';

        mkl_scsrmm(&transa, &m, &n, &m, &alpha, matdescra, values, columns, rowIndex, &(rowIndex[1]), &(b[0][0]), &n,  &beta, &(c[0][0]), &n);

        printf("                             \n");
        printf("   OUTPUT DATA FOR MKL_SCSRMM\n");
        for (i = 0; i < m; i++) {
            for (j = 0; j < n; j++) {
                printf("%7.1f", c[i][j]);
            };
            printf("\n");
        };
        return 0;
}

我得到的结果是:

从零开始的索引(右侧索引):

24.0 10.0

18.0 8.0

基于单一的索引:

8.0 10.0

18.0 24.0

虽然它似乎只改变了对角元素的位置,但是用3x3矩阵解决了它与正确的完全不同的解决方案。我怀疑它可能是矩阵b的输入格式。我认为对于放置在MKL参考手册中的函数mkl_scsrmm的数组b的描述缺乏清晰度。因此,我改变了b格式,在这个例子中它起作用了(我按照这个顺序放置元素位置:`{2.0,5.0,1.0,2.0})但是我对我编码的另一个3x3例子做了同样的事情并且它没有做到工作所以我认为这可能只是巧合。我真的不知道该怎么处理这个问题,我想了解这里发生了什么。

参考文献:

CSR格式

https://software.intel.com/en-us/node/471374

Spare Blas mkl_scsrmm功能

https://software.intel.com/sites/products/documentation/doclib/iss/2013/mkl/mklman/hh_goto.htm#GUID-78C55D9B-86FF-4A9F-B5D5-D2F61B9314FC.htm

备用Blas接口注意事项

https://software.intel.com/sites/products/documentation/doclib/iss/2013/mkl/mklman/hh_goto.htm#GUID-34C8DB79-0139-46E0-8B53-99F3BEE7B2D4.htm

这是另一个例子,3x3:

示例#2

//  matrix A
//
//  2   4   3  
//  4   2   1
//  3   1   6
//
//  matrix B
//
//  2   1   3
//  4   5   6
//  7   8   9
//  
//  ZERO-BASED INDEXING
//  
//  a = {2 4 3 4 2 1 3 1 6}
//  columns= {0 1 2 0 1 2 0 1 2}
//  idexRow = {0 3 6 9}
//
//  b = {2 1 3 4 5 6 7 8 9} (row order array)
//
//  We print the array in row-major order
//
//  ONE-BASED INDEXING
//  
//  a = {2 4 3 4 2 1 3 1 6}
//  columns={1 2 3 1 2 3 1 2 3}
//  indexRow = {0 3 6 9}
//
//  b = {2 4 7 1 5 8 3 6 9} (column order array)
//  
//  We print the array in column-major order (because the resoult is in column major order, ie transposed)
//
//
//

#include <stdio.h>
#include "mkl_types.h"
#include "mkl_spblas.h"

int main() 
{

#define M 3        
#define NNZ 9 
#define N 3 

        MKL_INT     m = M, nnz = NNZ, n=N;
        float       a[NNZ]    = {2.0,4.0,3.0,4.0,2.0,1.0,3.0,1.0,6.0};
        MKL_INT     columns[NNZ]  = {0,1,2,0,1,2,0,1,2};
        MKL_INT     rowIndex[M+1] = {0,3,6,9};

        float       b[M][N] = {2.0, 1.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0};
        float       c[M][N] = {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};

        float       alpha = 1.0, beta = 0.0;
        MKL_INT     i, j;
        char        transa;
        char        matdescra[6];

        float       a1[NNZ]   = {2.0,4.0,3.0,4.0,2.0,1.0,3.0,1.0,6.0};
        MKL_INT     columns1[NNZ]  = {1,2,3,1,2,3,1,2,3};
        MKL_INT     rowIndex1[M+1] = {1,4,7,10};
        float       b1[M][N]    = {2.0, 4.0, 7.0, 1.0, 5.0, 8.0, 3.0, 6.0, 9.0};
        float       c1[M][N]    = {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};

//********************************
        //ZERO-BASED INDEXING
//********************************

        transa = 'n';
        matdescra[0] = 's';
        matdescra[1] = 'l';
        matdescra[2] = 'n';
        matdescra[3] = 'c';

        mkl_scsrmm(&transa, &m, &n, &m, &alpha, matdescra, a, columns, rowIndex, &(rowIndex[1]), &(b[0][0]), &n,  &beta, &(c[0][0]), &n);

        printf("                             \n");
        printf("   Right Solution: ZERO-BASED: C    \n");
        for (i = 0; i < m; i++)
        {
            for (j = 0; j < n; j++) {
                printf("%7.1f", c[i][j]);
            };
            printf("\n");
        };

        printf("                             \n");
        printf("   ZERO-BASED: C'    \n");
        for (i = 0; i < m; i++) 
        {
            for (j = 0; j < n; j++) 
            {
                printf("%7.1f", c[j][i]);
            };
            printf("\n");
        };

//********************************
        //ONE-BASED INDEXING
//********************************

        matdescra[3] = 'f';

        mkl_scsrmm(&transa, &m, &n, &m, &alpha, matdescra, a1, columns1, rowIndex1, &(rowIndex1[1]), &(b1[0][0]), &n,  &beta, &(c1[0][0]), &n);

        printf("                             \n");
        printf("   ONE-BASED: C    \n");
        for (i = 0; i < m; i++) 
        {
            for (j = 0; j < n; j++) 
            {
                printf("%7.1f", c1[i][j]);
            };
            printf("\n");
        };

        printf("                             \n");
        printf("   ONE-BASED: C'    \n");
        for (i = 0; i < m; i++) 
        {
            for (j = 0; j < n; j++) 
            {
                printf("%7.1f", c1[j][i]);
            };
            printf("\n");
        };
        return 0;
}

1 个答案:

答案 0 :(得分:1)

我在英特尔论坛上问了同样的问题,我在那里得到了一些帮助,并找到问题的解决方案。这笔交易是当使用从零开始索引从C接口调用例程时,您可以按照 row-major 顺序发送存储在数组中的矩阵( native C数组存储),当您使用基于一个索引调用例程时,您必须以 column-major 顺序存储矩阵。这改变了矩阵B和C需要存储的方式,以及结果的存储方式。对于矩阵A,它只改变索引(从0到1)。从英特尔的文档中,您可能认为C接口始终接受两种索引类型的行主要排序。

请注意,一般来说,列主要排序 相同,因为将转置矩阵存储在行中 - 主要排序(如果矩阵是正方形,则相同)。