Commit 83460626 authored by D'azevedo, Ed's avatar D'azevedo, Ed
Browse files

use magma_Xgemm_vbatch instead of magma_dgemm_vbatch

parent 61a2cba9

src/dmrg_vbatch.c

+3 −3
Original line number Diff line number Diff line
@@ -204,7 +204,7 @@ double elapsed_time = 0; 


  /*

   ----------------------------------------------------------

   Note magma_dgemm_vbatched need arrays of size batch_size+1

   Note magma_Xgemm_vbatched need arrays of size batch_size+1

   Set vbatch_dim to be multiple of 32 for better alignment in memory

   ----------------------------------------------------------

   */
@@ -294,7 +294,7 @@ double elapsed_time = 0; 
   {

   /*

    ---------------------------------------------------------------

    Note magma_dgemm_vbatched assumes all transa, transb, alpha, beta are the same

    Note magma_Xgemm_vbatched assumes all transa, transb, alpha, beta are the same

    ---------------------------------------------------------------

    */

   int is_same_transa = 0;
@@ -357,7 +357,7 @@ double elapsed_time = 0; 


     



   magmablas_dgemm_vbatched( transA, transB,

   magmablas_Xgemm_vbatched( transA, transB,

           m_vbatch, n_vbatch, k_vbatch,

           alpha,

           (FpType const * const *) a_vbatch, lda_vbatch,