Ошибка Intel MKL: параметр 9 был неверным при входе в cblas_zgemm (C # Linux) - PullRequest
0 голосов
/ 06 марта 2019

Я создал пользовательскую библиотеку MKL (обновление 2019 для 2) для Windows (10) и Linux (Ubuntu 18.04), используя следующие команды:

nmake libintel64 MKLROOT="C:\Program Files (x86)\IntelSWTools\compilers_and_libraries\windows\mkl" name=win\intel64\custom_mkl interface="lp64"

и

make libintel64 MKLROOT="opt/intel/mkl" name=linux/intel64/custom_mkl interface=lp64

Я использую DllImport из C # для вызова cblas_zgemm

[DllImport(DLLName, CallingConvention = CallingConvention.Cdecl, ExactSpelling = true, SetLastError = false)]
internal static extern void cblas_zgemm(
    int Order, int TransA, int TransB, int M, int N, int K,
    ComplexDouble alpha, [In] ComplexDouble[,] A, int lda, [In] ComplexDouble[,] B, int ldb,
    ComplexDouble beta, [In, Out] ComplexDouble[,] C, int ldc);

см. https://software.intel.com/en-us/mkl-developer-reference-c-cblas-gemm

где ComplexDoubleопределяется как:

public struct ComplexDouble
{
    public double real;
    public double imag;

    public static implicit operator ComplexDouble(double d)
    {
        return new ComplexDouble() { real = d };
    }
}

Я определил следующий статический метод для умножения матриц:

public static ComplexDouble[,] Dot(ComplexDouble[,] a, ComplexDouble[,] b)
{
    int n1 = a.GetLength(0);
    int n2 = a.GetLength(1);
    int n3 = b.GetLength(0);
    int n4 = b.GetLength(1);
    if (n2 != n3) throw new System.Exception("Inner matrix dimensions must agree");
    int Order = 101; // row-major arrays
    int TransA = 111; // trans='N'
    int TransB = 111; // trans='N'
    int M = n1, N = n4, K = n2;
    int lda = K, ldb = N, ldc = N;
    ComplexDouble alpha = 1, beta = 0;
    ComplexDouble[,] c = new ComplexDouble[n1, n4];
    _mkl.cblas_zgemm(Order, TransA, TransB, M, N, K, alpha, a, lda, b, ldb, beta, c, ldc);
    return c;
}

И, наконец, у меня есть следующий тестовый код, который отлично работает под Windows, ноне работает в Linux (параметр 9 был неверным при входе в cblas_zgemm).

ComplexDouble[,] A = new ComplexDouble[,] { { 1, 2, 3 }, { 4, 5, 6 } };
ComplexDouble[,] B = new ComplexDouble[,] { { 0, 1, 0, 1 }, { 1, 0, 0, 1 }, { 1, 0, 1, 0 } };
ComplexDouble[,] C = MKL.Dot(A, B);

Я сделал следующие наблюдения:

  • Он работает в Windows
  • В Linux (Mono) происходит сбой; параметр 9 был неверным при входе в cblas_zgemm или Null Reference.
  • Если я повторю эксперимент с cblas_dgemm и double, он будет работать в Windows и в Linux.

1 Ответ

0 голосов
/ 07 марта 2019

При более внимательном рассмотрении cblas_zgemm

void cblas_zgemm (
    const CBLAS_LAYOUT Layout, const CBLAS_TRANSPOSE transa, const CBLAS_TRANSPOSE transb, const MKL_INT m, const MKL_INT n, const MKL_INT k, 
    const void *alpha, const void *a, const MKL_INT lda, const void *b, const MKL_INT ldb, 
    const void *beta, void *c, const MKL_INT ldc);

видно, что аргументы alpha и beta должны передаваться по ссылке.Поэтому DllImport должно выглядеть так:

[DllImport(DLLName, CallingConvention = CallingConvention.Cdecl, ExactSpelling = true, SetLastError = false)]
internal static extern void cblas_zgemm(
    int Order, int TransA, int TransB, int M, int N, int K,
    ref ComplexDouble alpha, [In] ComplexDouble[,] A, int lda, [In] ComplexDouble[,] B, int ldb,
    ref ComplexDouble beta, [In, Out] ComplexDouble[,] C, int ldc);

Я почему-то пропустил это, потому что для cblas_dgemm

void cblas_dgemm (
    const CBLAS_LAYOUT Layout, const CBLAS_TRANSPOSE transa, const CBLAS_TRANSPOSE transb, const MKL_INT m, const MKL_INT n, const MKL_INT k, 
    const double alpha, const double *a, const MKL_INT lda, const double *b, const MKL_INT ldb, 
    const double beta, double *c, const MKL_INT ldc);

аргументы alpha и beta передаются по значению.

[DllImport(DLLName, CallingConvention = CallingConvention.Cdecl, ExactSpelling = true, SetLastError = false)]
internal static extern void cblas_dgemm(
    int Order, int TransA, int TransB, int M, int N, int K,
    double alpha, [In] double[,] A, int lda, [In] double[,] B, int ldb,
    double beta, [In, Out] double[,] C, int ldc);

Вот так cblas_zgemm работает в Windows и Linux.Я до сих пор не понимаю, почему он работал на Windows с ref?

...