@@ -40,10 +40,10 @@ typedef float BaseFloat;
4040#endif
4141
4242
43- void cudaD_add_row_sum_mat (int Gr , int Bl , double * result , const double * mat ,
43+ void cudaD_add_row_sum_mat (double * result , const double * mat , void * scratch ,
4444 const MatrixDim d , const double alpha ,
4545 const double beta );
46- void cudaF_add_row_sum_mat (int Gr , int Bl , float * result , const float * mat ,
46+ void cudaF_add_row_sum_mat (float * result , const float * mat , void * scratch ,
4747 const MatrixDim d , const float alpha ,
4848 const float beta );
4949void cudaD_add_col_sum_mat (int Gr , int Bl , double * result , const double * mat ,
@@ -713,10 +713,10 @@ void cudaD_trace_mat_mat(dim3 Gr, dim3 Bl, const double* A, const double* B,
713713 MatrixDim dA , int B_stride , double * value );
714714void cudaF_trace_mat_mat (dim3 Gr , dim3 Bl , const float * A , const float * B ,
715715 MatrixDim dA , int B_stride , float * value );
716- void cudaD_trace_mat_mat_trans (dim3 Gr , dim3 Bl , const double * A ,
716+ void cudaD_trace_mat_mat_trans (const double * A ,
717717 const double * B , MatrixDim dA , int B_stride ,
718718 double * value );
719- void cudaF_trace_mat_mat_trans (dim3 Gr , dim3 Bl , const float * A , const float * B ,
719+ void cudaF_trace_mat_mat_trans (const float * A , const float * B ,
720720 MatrixDim dA , int B_stride , float * value );
721721void cudaD_trace_mat_smat (dim3 Gr , dim3 Bl , const double * mat ,
722722 MatrixDim mat_dim , const int * smat_row_ptr ,
0 commit comments