Skip to content

Commit bf36cb3

Browse files
committed
fix coverage yet
1 parent db89b5a commit bf36cb3

File tree

2 files changed

+6
-20
lines changed

2 files changed

+6
-20
lines changed

tasks/all/example/src/ops_all.cpp

+3-10
Original file line numberDiff line numberDiff line change
@@ -9,8 +9,7 @@
99
#include <vector>
1010

1111
#include "core/util/include/util.hpp"
12-
#include "oneapi/tbb/task_arena.h"
13-
#include "oneapi/tbb/task_group.h"
12+
#include "oneapi/tbb/parallel_for.h"
1413

1514
void nesterov_a_test_task_all::MatMul(const std::vector<int> &in_vec, int rc_size, std::vector<int> &out_vec) {
1615
for (int i = 0; i < rc_size; ++i) {
@@ -24,14 +23,8 @@ void nesterov_a_test_task_all::MatMul(const std::vector<int> &in_vec, int rc_siz
2423
}
2524

2625
void nesterov_a_test_task_all::MatMulTBB(const std::vector<int> &in_vec, int rc_size, std::vector<int> &out_vec) {
27-
oneapi::tbb::task_arena arena(1);
28-
arena.execute([&] {
29-
tbb::task_group tg;
30-
for (int i = 0; i < ppc::util::GetPPCNumThreads(); ++i) {
31-
tg.run([&] { MatMul(in_vec, rc_size, out_vec); });
32-
}
33-
tg.wait();
34-
});
26+
tbb::parallel_for(0, ppc::util::GetPPCNumThreads(), [&](int i) { MatMul(in_vec, rc_size - i, out_vec); });
27+
MatMul(in_vec, rc_size, out_vec);
3528
}
3629

3730
bool nesterov_a_test_task_all::TestTaskALL::ValidationImpl() {

tasks/tbb/example/src/ops_tbb.cpp

+3-10
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
#include <cstddef>
88
#include <vector>
99

10-
#include "oneapi/tbb/task_arena.h"
11-
#include "oneapi/tbb/task_group.h"
10+
#include "oneapi/tbb/parallel_for.h"
1211

1312
namespace {
1413
void MatMul(const std::vector<int> &in_vec, int rc_size, std::vector<int> &out_vec) {
@@ -35,14 +34,8 @@ bool nesterov_a_test_task_tbb::TestTaskTBB::PreProcessingImpl() {
3534
}
3635

3736
bool nesterov_a_test_task_tbb::TestTaskTBB::RunImpl() {
38-
oneapi::tbb::task_arena arena(1);
39-
arena.execute([&] {
40-
tbb::task_group tg;
41-
for (int thr = 0; thr < ppc::util::GetPPCNumThreads(); ++thr) {
42-
tg.run([&] { MatMul(input_, rc_size_, output_); });
43-
}
44-
tg.wait();
45-
});
37+
tbb::parallel_for(0, ppc::util::GetPPCNumThreads(), [&](int i) { MatMul(input_, rc_size_ - i, output_); });
38+
MatMul(input_, rc_size_, output_);
4639
return true;
4740
}
4841

0 commit comments

Comments
 (0)