|
|
|
@@ -328,7 +328,11 @@ int MatmulCPUKernel::Run() { |
|
|
|
cur_b_ptr_ = b_ptr_ + i * params_->deep_ * params_->col_; |
|
|
|
cur_c_ptr_ = c_src + i * params_->row_ * params_->col_; |
|
|
|
} else { |
|
|
|
#if defined(ENABLE_ARM32) || defined(ENABLE_X86_64_SSE) |
|
|
|
cur_a_ptr_ = a_ptr_ + i * params_->row_4_ * params_->deep_; |
|
|
|
#else |
|
|
|
cur_a_ptr_ = a_ptr_ + i * params_->row_12_ * params_->deep_; |
|
|
|
#endif |
|
|
|
cur_b_ptr_ = b_ptr_ + i * params_->deep_ * params_->col_8_; |
|
|
|
cur_c_ptr_ = c_src + i * params_->row_ * params_->col_; |
|
|
|
} |
|
|
|
|