Ticket #14653: _dotblas.diff

File _dotblas.diff, 2.6 KB (added by MarcusCalhoun-Lopez (Marcus Calhoun-Lopez), 17 years ago)

  • Packages/dotblas/dotblas/_dotblas.c

    old new  
    253253                  ap1->dimensions[0], ap2->dimensions[1], ap2->dimensions[0],
    254254                  1.0, (double *)ap1->data, lda,
    255255                  (double *)ap2->data, ldb,
    256                   0.0, (double *)ret->data, ldb);
     256                  0.0, (double *)ret->data, ret->dimensions[1]);
    257257    }
    258258    else if (typenum == PyArray_FLOAT) {
    259259      cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans,
    260260                  ap1->dimensions[0], ap2->dimensions[1], ap2->dimensions[0],
    261261                  1.0, (float *)ap1->data, lda,
    262262                  (float *)ap2->data, ldb,
    263                   0.0, (float *)ret->data, ldb);
     263                  0.0, (float *)ret->data, ret->dimensions[1]);
    264264    }
    265265    else if (typenum == PyArray_CDOUBLE) {
    266266      cblas_zgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans,
    267267                  ap1->dimensions[0], ap2->dimensions[1], ap2->dimensions[0],
    268268                  oneD, (double *)ap1->data, lda,
    269269                  (double *)ap2->data, ldb,
    270                   zeroD, (double *)ret->data, ldb);
     270                  zeroD, (double *)ret->data, ret->dimensions[1]);
    271271    }
    272272    else if (typenum == PyArray_CFLOAT) {
    273273      cblas_cgemm(CblasRowMajor, CblasNoTrans, CblasNoTrans,
    274274                  ap1->dimensions[0], ap2->dimensions[1], ap2->dimensions[0],
    275275                  oneF, (float *)ap1->data, lda,
    276276                  (float *)ap2->data, ldb,
    277                   zeroF, (float *)ret->data, ldb);
     277                  zeroF, (float *)ret->data, ret->dimensions[1]);
    278278    }
    279279  }
    280280  else {
    … …  
    530530                  ap1->dimensions[0], ap2->dimensions[0], ap1->dimensions[1],
    531531                  1.0, (double *)ap1->data, lda,
    532532                  (double *)ap2->data, ldb,
    533                   0.0, (double *)ret->data, ldb);
     533                  0.0, (double *)ret->data, ret->dimensions[1]);
    534534    }
    535535    else if (typenum == PyArray_FLOAT) {
    536536      cblas_sgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
    537537                  ap1->dimensions[0], ap2->dimensions[0], ap1->dimensions[1],
    538538                  1.0, (float *)ap1->data, lda,
    539539                  (float *)ap2->data, ldb,
    540                   0.0, (float *)ret->data, ldb);
     540                  0.0, (float *)ret->data, ret->dimensions[1]);
    541541    }
    542542    else if (typenum == PyArray_CDOUBLE) {
    543543      cblas_zgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
    544544                  ap1->dimensions[0], ap2->dimensions[0], ap1->dimensions[1],
    545545                  oneD, (double *)ap1->data, lda,
    546546                  (double *)ap2->data, ldb,
    547                   zeroD, (double *)ret->data, ldb);
     547                  zeroD, (double *)ret->data, ret->dimensions[1]);
    548548    }
    549549    else if (typenum == PyArray_CFLOAT) {
    550550      cblas_cgemm(CblasRowMajor, CblasNoTrans, CblasTrans,
    551551                  ap1->dimensions[0], ap2->dimensions[0], ap1->dimensions[1],
    552552                  oneF, (float *)ap1->data, lda,
    553553                  (float *)ap2->data, ldb,
    554                   zeroF, (float *)ret->data, ldb);
     554                  zeroF, (float *)ret->data,  ret->dimensions[1]);
    555555    }
    556556  }
    557557  else {