saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-14 20:48:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use #pragma parallel for properly (#68)

Browse source
This commit is contained in:
Tracy Sharpe 2018-11-30 14:11:49 -08:00 committed by GitHub
parent c55663860e
commit 21044846ee
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/core/mlas/lib/sgemm.cpp
View file

@ -1288,9 +1288,8 @@ Return Value:
#if defined(MLAS_USE_OPENMP)
#pragma omp parallel num_threads(Index)
{
int tid = omp_get_thread_num();
#pragma omp parallel for
for (int32_t tid = 0; tid < int32_t(Index); tid++) {
MLAS_SGEMM_WORK_BLOCK::SEGMENT* Segment = &WorkBlock.Segments[tid];