Blas dgemv медленнее, чем ручная реализация

Я пытаюсь использовать BLAS в моей c-программе, чтобы улучшить скорость матричного / векторного произведения.

Вручную у меня был этот код:

for (j = 0; j < ann->hidden; ++j) {

        double delta = 0;
        //h is known before 
        for (k = 0; k < (h == ann->hidden_layers-1 ? ann->outputs : ann->hidden); ++k) {
            const double forward_delta = dd[k];
            const int windex = k * (ann->hidden + 1) + (j + 1);
            const double forward_weight = ww[windex];
            delta += forward_delta * forward_weight;
        }
        *d = *o * (1.0-*o) * delta;
        ++d; ++o;
    }
}

Итак, я попытался заменить этот двойник на функцию blas cblas_dgemv, и это выглядит так:

int n = h == ann->hidden_layers-1 ? ann->outputs : ann->hidden ;
    int m = ann->hidden ;
    double *delta = calloc(m,sizeof(double));
    cblas_dgemv(CblasColMajor,CblasNoTrans,m,n,1,&ww[1],m,dd,1,0.0,delta,1);
    for(j=0 ;  j < ann->hidden; ++j) {
      *d = *o * (1.0-*o)*delta[j];
      ++d; ++o;

    }
    free(delta);
    }

Выходные значения хороши.

Проблема в том, что моя реализация с BLAS намного медленнее, чем "ручная"...

Я не знаю, потому что я не использую наиболее оптимизированную функцию для этого расчета или я сделал что-то не так?

1 ответ

Решение

Хорошо, дело в том, что моя реализация blas быстрее для большой матрицы, но издержки blas замедляют работу для маленькой матрицы!

Другие вопросы по тегам