| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | gemm_interleaved_pretransposed_2d.hpp | 68 const int _maxthreads; member in arm_gemm::GemmInterleavedPretransposed2d
203 auto a_panel_start = c_panel_start + get_c_working_size() * _maxthreads; in execute_pretranspose()
349 , _maxthreads(args._maxthreads) in GemmInterleavedPretransposed2d()
350 , _nthreads(args._maxthreads) in GemmInterleavedPretransposed2d()
359 assert(_maxthreads > 0); in GemmInterleavedPretransposed2d()
398 _nthreads = std::min(nthreads, _maxthreads); in set_nthreads()
427 return get_c_working_size() * _maxthreads in get_working_size()
428 + get_a_working_size() * _maxthreads in get_working_size()
524 … unsigned int ideal_height = static_cast<unsigned int>(std::sqrt(args._maxthreads * ratio) + 0.5); in estimate_cycles()
532 if (args._maxthreads % round_down == 0) { in estimate_cycles()
[all …]
|
| H A D | gemm_interleaved.hpp | 347 const int _maxthreads; member in arm_gemm::GemmInterleaved
467 … return ROUND_UP(sizeof(Toi) * get_total_k_depth() * strategy::out_height() * _maxthreads); in get_a_working_size()
546 if (args._maxthreads == 1) { in is_thread_columns()
554 if (args._maxthreads > m_blocks) { in is_thread_columns()
559 if (((roundup(m_blocks, args._maxthreads) * 100) / m_blocks) > 120) { in is_thread_columns()
676 _act(args._act), _maxthreads(args._maxthreads), _nthreads(args._maxthreads), in GemmInterleaved()
686 _act(args._act), _maxthreads(args._maxthreads), _nthreads(args._maxthreads), in GemmInterleaved()
709 _nthreads = std::min(nthreads, _maxthreads); in set_nthreads()
747 …Toi * const a_panel = reinterpret_cast<Toi *>(working_space_bytes + (_maxthreads * get_c_working_s… in execute()
838 …Toi * const a_panel = reinterpret_cast<Toi *>(working_space_bytes + (_maxthreads * get_c_working_s… in execute()
[all …]
|
| H A D | quantize_wrapper.hpp | 87 unsigned int first_row = (threadid * _args._Msize) / _args._maxthreads; in requantize_runtime()
88 unsigned int last_row = ((threadid+1) * _args._Msize) / _args._maxthreads; in requantize_runtime()
112 …st GemmArgs &args, const Requantize32 &qp) : _params(qp), _args(args), _barrier(args._maxthreads) { in QuantizeWrapper()
113 …gs._Ksections, args._nbatches, args._nmulti, args._indirect_input, Activation(), args._maxthreads); in QuantizeWrapper()
138 _args._maxthreads = nthreads; in set_nthreads()
|
| H A D | gemm_quint8.cpp | 80 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon2c615aac0302()
88 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon2c615aac0602()
96 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon2c615aac0902()
|
| H A D | gemm_qint8.cpp | 84 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon1bcad4d70302()
92 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon1bcad4d70602()
100 …s &args, const Requantize32 &qp) { return args._ci->has_sme2() && args._maxthreads == 1 && ((qp.pe… in __anon1bcad4d70902()
|
| H A D | gemm_hybrid_indirect.hpp | 364 if (multi_row_parallelism < args._maxthreads) {
365 unsigned int columns_needed = iceildiv(args._maxthreads, multi_row_parallelism);
377 if (args._Ksize <= 128 && args._maxthreads <= 16) {
|
| H A D | gemm_hybrid_quantized_inline.hpp | 146 _qp (qp), _nthreads(args._maxthreads) { } in GemmHybridQuantizedInline()
|
| H A D | gemm_hybrid.hpp | 118 if ((args._Ksize <= 128) && (args._maxthreads <= 16)) { in compute_n_block()
|
| H A D | gemm_hybrid_quantized.hpp | 162 _qp (qp), _nthreads(args._maxthreads) { } in GemmHybridQuantized()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/assembly/ |
| H A D | arm_gemm.hpp | 154 int _maxthreads; member
163 …es(nbatches), _nmulti(nmulti), _indirect_input(indirect_input), _act(act), _maxthreads(maxthreads), in GemmArgs()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/ |
| H A D | CpuGemmAssemblyDispatch.cpp | 388 if(window_size < static_cast<unsigned int>(args._maxthreads)) in configure()
|