0

「cblas_dgemm()」の呼び出しを cublasDgemm() に置き換えたいと考えています。Shark 機械学習ライブラリの元のラッパーは次のとおりです。 

inline void gemm(
    CBLAS_ORDER const Order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
    int M, int N, int K,
    double alpha, double const *A, int lda,
    double const *B, int ldb,
    double beta, double *C, int ldc
){

    cblas_dgemm(
        Order, TransA, TransB,
        M, N, K,
        alpha, 
        A, lda,
        B, ldb,
        beta, 
        C, ldc
    );

}

OpenAcc プラグマを使用して変更されたコードは次のとおりです。

inline void gemm(
    CBLAS_ORDER const Order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
    int M, int N, int K,
    double alpha, double const *A, int lda,
    double const *B, int ldb,
    double beta, double *C, int ldc
){
        #ifdef _OPENACC
        cublasOperation_t OpT_A, OpT_B; 
            switch (TransA)
        {
            case CblasNoTrans:
                OpT_A = CUBLAS_OP_N;
                break;
            case CblasTrans:
                OpT_A = CUBLAS_OP_T;
                break;
            case CblasConjTrans:
                OpT_A = CUBLAS_OP_C;
                break;
            default:
                                OpT_A = CUBLAS_OP_N;
        }
               switch (TransB)
                {
                        case CblasNoTrans:
                                OpT_B = CUBLAS_OP_N;
                break;
                        case CblasTrans:
                                OpT_B = CUBLAS_OP_T;
                break;
                        case CblasConjTrans:
                                OpT_B = CUBLAS_OP_C;
                break;
                        default:
                                OpT_B = CUBLAS_OP_N;
                }

              cublasHandle_t handle;
              #pragma acc data copyin(OpT_A, OpT_B, M, N, K, alpha, A[0:M][0:K], lda, B[0:K][0:N], ldb, beta, ldc) copy(C[0:M][0:N])
                      {
                                 #pragma acc host_data use_device(handle,OpT_A, OpT_B, A, B, C, M, N, K, lda, ldb, ldc, alpha, beta)
                                 {
                                     cublasDgemm(handle,OpT_A,OpT_B,M,N,K,&alpha,A,lda,B,ldb,&beta,C,ldc);
                                }
                        }

    #else

    cblas_dgemm(
        Order, TransA, TransB,
        M, N, K,
        alpha, 
        A, lda,
        B, ldb,
        beta, 
        C, ldc
    );
    #endif
}

問題は、OpenAcc フラグを使用してコードをコンパイルすると、結果行列の要素、つまり C がカーネル実行の前後ですべてゼロになることです。ここで何が欠けているのかわかりません。助けていただければ幸いです。

4

1 に答える 1

1

基本的な構造が整っているようです。dataただし、 orhost_dataディレクティブのスカラー変数は必要ありません。それが Op*、M、N、K、および ld* 変数です。cublasDgemm はカーネルを起動するためにホスト上のこれらの変数を解決しようとするため、これはおそらくあなたの問題だと思います。

于 2015-12-10T15:09:50.520 に答える