/dports/science/hypre/hypre-2.23.0/src/parcsr_ls/ |
H A D | par_amg_setup.c | 36 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGSetup() local 272 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGSetup() 380 if (A_array[j]) in hypre_BoomerAMGSetup() 382 hypre_ParCSRMatrixDestroy(A_array[j]); in hypre_BoomerAMGSetup() 383 A_array[j] = NULL; in hypre_BoomerAMGSetup() 570 if (A_array == NULL) in hypre_BoomerAMGSetup() 642 A_array[0] = A; in hypre_BoomerAMGSetup() 670 hypre_ParAMGDataAArray(amg_data) = A_array; in hypre_BoomerAMGSetup() 1090 hypre_BoomerAMGCreateS(A_array[level], in hypre_BoomerAMGSetup() 2881 A_array[level] = A_H; in hypre_BoomerAMGSetup() [all …]
|
H A D | par_mgr_solve.c | 30 hypre_ParCSRMatrix **A_array = (mgr_data -> A_array); in hypre_MGRSolve() local 149 hypre_ParCSRMatrixMatvec(alpha, A_array[0], U_array[0], beta, Vtemp); in hypre_MGRSolve() 380 hypre_ParCSRMatrix **A_array = ((Frelax_data) -> A_array); in hypre_MGRFrelaxVcycle() local 417 Solve_err_flag = hypre_BoomerAMGRelaxIF(A_array[0], in hypre_MGRFrelaxVcycle() 434 Solve_err_flag = hypre_BoomerAMGRelax(A_array[0], in hypre_MGRFrelaxVcycle() 498 Solve_err_flag = hypre_BoomerAMGRelaxIF(A_array[level], in hypre_MGRFrelaxVcycle() 529 Solve_err_flag = hypre_BoomerAMGRelaxIF(A_array[level], in hypre_MGRFrelaxVcycle() 598 hypre_ParCSRMatrix **A_array = (mgr_data -> A_array); in hypre_MGRCycle() local 639 comm = hypre_ParCSRMatrixComm(A_array[0]); in hypre_MGRCycle() 740 hypre_ParCSRMatrixMatvecOutOfPlace(-1.0, A_array[level], in hypre_MGRCycle() [all …]
|
H A D | par_cycle.c | 32 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGCycle() local 112 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGCycle() 169 num_coeffs[0] = hypre_ParCSRMatrixDNumNonzeros(A_array[0]); in hypre_BoomerAMGCycle() 170 comm = hypre_ParCSRMatrixComm(A_array[0]); in hypre_BoomerAMGCycle() 184 num_coeffs[j] = hypre_ParCSRMatrixDNumNonzeros(A_array[j]); in hypre_BoomerAMGCycle() 307 hypre_ParCSRMatrixMatvecOutOfPlace(alpha, A_array[level], in hypre_BoomerAMGCycle() 459 (HYPRE_ParCSRMatrix) A_array[level], in hypre_BoomerAMGCycle() 466 (HYPRE_ParCSRMatrix) A_array[level], in hypre_BoomerAMGCycle() 497 A_array[level], in hypre_BoomerAMGCycle() 507 hypre_ParCSRRelax_Cheby_Solve(A_array[level], Aux_F, in hypre_BoomerAMGCycle() [all …]
|
H A D | par_amg_solveT.c | 42 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGSolveT() local 83 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGSolveT() 93 num_coeffs[0] = hypre_ParCSRMatrixDNumNonzeros(A_array[0]); in hypre_BoomerAMGSolveT() 94 num_variables[0] = hypre_ParCSRMatrixGlobalNumRows(A_array[0]); in hypre_BoomerAMGSolveT() 96 A_array[0] = A; in hypre_BoomerAMGSolveT() 111 num_coeffs[j] = hypre_ParCSRMatrixDNumNonzeros(A_array[j]); in hypre_BoomerAMGSolveT() 309 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGCycleT() local 360 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGCycleT() 389 num_coeffs[0] = hypre_ParCSRMatrixDNumNonzeros(A_array[0]); in hypre_BoomerAMGCycleT() 392 num_coeffs[j] = hypre_ParCSRMatrixDNumNonzeros(A_array[j]); in hypre_BoomerAMGCycleT() [all …]
|
H A D | par_amgdd_solve.c | 20 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGDDSolve() local 62 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGDDSolve() 96 A_array[0] = A; in hypre_BoomerAMGDDSolve() 99 if (A != A_array[0]) in hypre_BoomerAMGDDSolve() 123 hypre_ParCSRMatrixMatvec(alpha, A_array[0], U_array[0], beta, res); in hypre_BoomerAMGDDSolve() 132 hypre_ParCSRMatrixMatvec(alpha, A_array[0], U_array[0], beta, Vtemp); in hypre_BoomerAMGDDSolve() 214 hypre_ParCSRMatrixMatvec(alpha, A_array[amgdd_start_level], in hypre_BoomerAMGDDSolve() 286 hypre_ParCSRMatrixMatvecOutOfPlace(alpha, A_array[0], U_array[0], beta, in hypre_BoomerAMGDDSolve() 367 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGDD_ResidualCommunication() local 394 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGDD_ResidualCommunication() [all …]
|
H A D | par_mgr_setup.c | 68 hypre_ParCSRMatrix **A_array = (mgr_data -> A_array); in hypre_MGRSetup() local 326 if (A_array || P_array || RT_array || CF_marker_array) in hypre_MGRSetup() 330 if (A_array[j]) in hypre_MGRSetup() 332 hypre_ParCSRMatrixDestroy(A_array[j]); in hypre_MGRSetup() 333 A_array[j] = NULL; in hypre_MGRSetup() 522 if (A_array == NULL) in hypre_MGRSetup() 618 (mgr_data -> A_array) = A_array; in hypre_MGRSetup() 729 A_array[lev] = RAP_ptr; in hypre_MGRSetup() 805 hypre_BoomerAMGBuildRestrAIR(A_array[lev], CF_marker, in hypre_MGRSetup() 824 AP = hypre_ParMatmul(A_array[lev], P_array[lev]); in hypre_MGRSetup() [all …]
|
H A D | par_amg_solve.c | 44 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGSolve() local 93 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGSolve() 105 A_array[0] = A; in hypre_BoomerAMGSolve() 162 hypre_ParCSRMatrixMatvec(alpha, A_array[0], U_array[0], beta, Residual ); in hypre_BoomerAMGSolve() 171 hypre_ParCSRMatrixMatvec(alpha, A_array[0], U_array[0], beta, Vtemp); in hypre_BoomerAMGSolve() 267 … hypre_ParCSRMatrixMatvecOutOfPlace(alpha, A_array[0], U_array[0], beta, F_array[0], Residual ); in hypre_BoomerAMGSolve() 272 … hypre_ParCSRMatrixMatvecOutOfPlace(alpha, A_array[0], U_array[0], beta, F_array[0], Vtemp); in hypre_BoomerAMGSolve() 355 num_coeffs[j] = (HYPRE_Real) hypre_ParCSRMatrixNumNonzeros(A_array[j]); in hypre_BoomerAMGSolve() 356 num_variables[j] = (HYPRE_Real) hypre_ParCSRMatrixGlobalNumRows(A_array[j]); in hypre_BoomerAMGSolve()
|
H A D | par_add_cycle.c | 28 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGAdditiveCycle() local 81 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGAdditiveCycle() 345 hypre_ParCSRMatrix **A_array; in hypre_CreateLambda() local 477 A_tmp = A_array[i]; in hypre_CreateLambda() 516 A_tmp = A_array[i]; in hypre_CreateLambda() 572 A_tmp = A_array[i]; in hypre_CreateLambda() 614 A_tmp = A_array[i]; in hypre_CreateLambda() 780 A_tmp = A_array[level]; in hypre_CreateLambda() 1031 hypre_ParCSRMatrix **A_array; in hypre_CreateDinv() local 1085 A_tmp = A_array[i]; in hypre_CreateDinv() [all …]
|
H A D | par_amgdd_setup.c | 36 hypre_ParCSRMatrix **A_array; in hypre_BoomerAMGDDSetup() local 77 A_array = hypre_ParAMGDataAArray(amg_data); in hypre_BoomerAMGDDSetup() 136 hypre_BoomerAMGDD_SetupNearestProcessorNeighbors(A_array[level], in hypre_BoomerAMGDDSetup() 147 comm = hypre_ParCSRMatrixComm(A_array[level]); in hypre_BoomerAMGDDSetup() 163 comm = hypre_ParCSRMatrixComm(A_array[level]); in hypre_BoomerAMGDDSetup()
|
/dports/math/libxsmm/libxsmm-1.16.3/src/template/ |
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_core.tpl.c | 249 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 256 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 263 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 270 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 410 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 417 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dci, in, ik, K); 423 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, df, in, ik, K); 429 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dp, in, ik, K); 443 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 456 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, df, in, ik, K); [all …]
|
H A D | libxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck_fused.tpl.c | 28 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wi, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 33 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 46 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, ri, ikb, icb + CB*KB_BLOCKS, 0, 0, kBlocks, bk, bk); 51 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, ri, ikb, icb + CB*KB_BLOCKS, 0, 0, kBlocks, bk, bk); 57 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 71 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wc, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 76 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, ci, j, in, ik, N, K), &blocks); 100 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, ci, j, in, ik, N, K), &blocks); 119 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); 143 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); [all …]
|
H A D | libxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck_diffused.tpl.c | 28 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wi, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 33 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 48 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wc, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 53 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, ci, j, in, ik, N, K), &blocks); 68 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wf, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 73 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); 88 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wo, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 93 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, o, j, in, ik, N, K), &blocks); 130 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 180 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); [all …]
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c | 126 const element_filter_type *A_array[1024]; variable 450 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 457 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 475 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 481 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 568 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 575 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dc, in, ik, K); 581 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, df, in, ik, K); 595 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 602 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dc, in, ik, K); [all …]
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c | 141 const element_filter_type *A_array[1024]; variable 430 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 437 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks); 455 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 461 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 548 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 555 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dc, in, ik, K); 561 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, df, in, ik, K); 575 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, di, in, ik, K); 582 A_array[inb] = &LIBXSMM_VLA_ACCESS(2, dc, in, ik, K); [all …]
|
H A D | libxsmm_dnn_rnncell_st_gru_fwd_nc_kcck.tpl.c | 64 const element_filter_type *A_array[1024]; variable 118 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wi, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 123 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 138 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 143 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wc, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 148 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, c, j, in, ik, N, K), &blocks); 163 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, c, j, in, ik, N, K), &blocks); 188 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wf, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 193 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); 196 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, rf, ikb, icb + CB*KB_BLOCKS, 0, 0, kBlocks, bk, bk); [all …]
|
H A D | libxsmm_dnn_rnncell_st_gru_fwd_nc_ck_generic.tpl.c | 79 const element_filter_type *A_array[1024]; variable 181 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wi, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 186 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 201 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, i, j, in, ik, N, K), &blocks); 206 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wc, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 211 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, c, j, in, ik, N, K), &blocks); 226 batchreduce_kernelb(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, c, j, in, ik, N, K), &blocks); 251 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, wf, ikb, icb + CB*CB_BLOCKS, 0, 0, cBlocks, bc, bk); 256 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, f, j, in, ik, N, K), &blocks); 259 A_array[icb] = &LIBXSMM_VLA_ACCESS(4, rf, ikb, icb + CB*KB_BLOCKS, 0, 0, kBlocks, bk, bk); [all …]
|
H A D | libxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c | 131 const element_filter_type *A_array[1024]; variable 231 A_array[ikb] = &LIBXSMM_VLA_ACCESS(4, wT, icb, ikb + KB*KB_BLOCKS, 0, 0, kBlocks, bk, bc); 236 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, dx, t-1, in, ic, N, C), &blocks); 250 A_array[inb] = &LIBXSMM_VLA_ACCESS(3, delta, t-1, in, ik, N, K); 265 A_array[inb] = &LIBXSMM_VLA_ACCESS(3, delta, t-1, in, ik, N, K); 284 A_array[ic] = &LIBXSMM_VLA_ACCESS(4, rT, ikb, ic, 0, 0, kBlocks, bk, bk); 289 batchreduce_kerneld(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, delta, i, in, ik, N, K) , &blocks); 358 A_array[ikb] = &LIBXSMM_VLA_ACCESS(4, wT, icb, ikb + KB*KB_BLOCKS, 0, 0, kBlocks, bk, bc); 363 batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, dx, i, in, ic, N, C), &blocks); 379 A_array[inb] = &LIBXSMM_VLA_ACCESS(3, delta, i, in, ik, N, K); [all …]
|
H A D | libxsmm_dnn_rnncell_st_rnn_fwd_ncnc_kcck.tpl.c | 73 const element_input_type *A_array[16][2][32]; variable 94 A_array[ii][jj][ic] = &LIBXSMM_VLA_ACCESS(4, w, ik, ic, 0, 0, cBlocks, bc, bk); 119 const element_input_type **src_ptr = &A_array[0][0][0]; 149 …batchreduce_kernela(&A_array[ii][jj][0], &B_array[ii][jj][0], &LIBXSMM_VLA_ACCESS(5, z, i, in, ik,… 169 const element_input_type *A_array[1024]; variable 193 A_array[ic] = &LIBXSMM_VLA_ACCESS(4, w, ik, ic, 0, 0, cBlocks, bc, bk); 198 …batchreduce_kernela(A_array, B_array, &LIBXSMM_VLA_ACCESS(5, z, i, in, ik, 0, 0, nBlocks, kBlocks,…
|
/dports/science/hypre/hypre-2.23.0/src/struct_ls/ |
H A D | sparse_msg_solve.c | 41 hypre_StructMatrix **A_array = (smsg_data -> A_array); in hypre_SparseMSGSolve() local 89 hypre_StructMatrixDestroy(A_array[0]); in hypre_SparseMSGSolve() 92 A_array[0] = hypre_StructMatrixRef(A); in hypre_SparseMSGSolve() 151 hypre_PFMGRelax(relax_array[0], A_array[0], b_array[0], x_array[0]); in hypre_SparseMSGSolve() 157 -1.0, A_array[0], x_array[0], 1.0, r_array[0]); in hypre_SparseMSGSolve() 232 hypre_PFMGRelax(relax_array[fi], A_array[fi], b_array[fi], in hypre_SparseMSGSolve() 238 -1.0, A_array[fi], x_array[fi], in hypre_SparseMSGSolve() 332 hypre_PFMGRelax(relax_array[fi], A_array[fi], b_array[fi], in hypre_SparseMSGSolve() 423 hypre_PFMGRelax(relax_array[fi], A_array[fi], b_array[fi], in hypre_SparseMSGSolve() 450 hypre_PFMGRelax(relax_array[0], A_array[0], b_array[0], x_array[0]); in hypre_SparseMSGSolve()
|
/dports/science/py-scipy/scipy-1.7.1/scipy/sparse/linalg/tests/ |
H A D | test_interface.py | 269 for M, A_array in self.cases: 285 assert_equal(A.matvec(x), A_array.dot(x)) 286 assert_equal(A * x, A_array.dot(x)) 288 assert_equal(A.matmat(x2), A_array.dot(x2)) 289 assert_equal(A * x2, A_array.dot(x2)) 293 assert_equal(A.T.matvec(y), A_array.T.dot(y)) 300 assert_equal(A.T.matmat(y), A_array.T.dot(y)) 310 for M, A_array in self.cases: 318 assert_equal(A.dot(x0), A_array.dot(x0)) 319 assert_equal(A.dot(x1), A_array.dot(x1)) [all …]
|
/dports/misc/tvm/incubator-tvm-0.6.1/src/runtime/contrib/cblas/ |
H A D | cblas.cc | 79 std::vector<const float*> A_array(batch_size); in operator ()() local 83 A_array[i] = A + i * a_stride; in operator ()() 87 cblas_sgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()() 124 std::vector<const double*> A_array(batch_size); in operator ()() local 128 A_array[i] = A + i * a_stride; in operator ()() 132 cblas_dgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()()
|
/dports/misc/py-tvm/incubator-tvm-0.6.1/src/runtime/contrib/cblas/ |
H A D | cblas.cc | 79 std::vector<const float*> A_array(batch_size); in operator ()() local 83 A_array[i] = A + i * a_stride; in operator ()() 87 cblas_sgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()() 124 std::vector<const double*> A_array(batch_size); in operator ()() local 128 A_array[i] = A + i * a_stride; in operator ()() 132 cblas_dgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cblas/ |
H A D | mkl.cc | 91 std::vector<const float*> A_array(batch_size); in operator ()() local 95 A_array[i] = A + i * a_stride; in operator ()() 99 cblas_sgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()() 127 std::vector<const double*> A_array(batch_size); in operator ()() local 131 A_array[i] = A + i * a_stride; in operator ()() 135 cblas_dgemm_batch(CblasColMajor, &trans_a, &trans_b, &M, &N, &K, &alpha, A_array.data(), &lda, in operator ()()
|
/dports/science/hypre/hypre-2.23.0/src/FEI_mv/femli/ |
H A D | mli_solver_hsgs.cxx | 165 amgData->A_array = new hypre_ParCSRMatrix*[1]; in calcOmega() 166 amgData->A_array[0] = A; in calcOmega() 176 delete [] amgData->A_array; in calcOmega()
|
H A D | mli_solver_hschwarz.cxx | 139 amgData->A_array = new hypre_ParCSRMatrix*[1]; in calcOmega() 140 amgData->A_array[0] = A; in calcOmega() 169 delete [] amgData->A_array; in calcOmega()
|