Home
last modified time | relevance | path

Searched refs:A_array (Results 1 – 25 of 48) sorted by relevance

12

/dports/science/hypre/hypre-2.23.0/src/parcsr_ls/
H A Dpar_amg_setup.c36 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 Dpar_mgr_solve.c30 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 Dpar_cycle.c32 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 Dpar_amg_solveT.c42 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 Dpar_amgdd_solve.c20 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 Dpar_mgr_setup.c68 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 Dpar_amg_solve.c44 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 Dpar_add_cycle.c28 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 Dpar_amgdd_setup.c36 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 Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_core.tpl.c249 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 Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck_fused.tpl.c28 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 Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck_diffused.tpl.c28 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 Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c126 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 Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c141 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 Dlibxsmm_dnn_rnncell_st_gru_fwd_nc_kcck.tpl.c64 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 Dlibxsmm_dnn_rnncell_st_gru_fwd_nc_ck_generic.tpl.c79 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 Dlibxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c131 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 Dlibxsmm_dnn_rnncell_st_rnn_fwd_ncnc_kcck.tpl.c73 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 Dsparse_msg_solve.c41 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 Dtest_interface.py269 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 Dcblas.cc79 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 Dcblas.cc79 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 Dmkl.cc91 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 Dmli_solver_hsgs.cxx165 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 Dmli_solver_hschwarz.cxx139 amgData->A_array = new hypre_ParCSRMatrix*[1]; in calcOmega()
140 amgData->A_array[0] = A; in calcOmega()
169 delete [] amgData->A_array; in calcOmega()

12