Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
项目经理老王
Mace
提交
cabe5cbe
Mace
项目概览
项目经理老王
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Mace
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
cabe5cbe
编写于
4月 03, 2019
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'mthreadpool' into 'master'
Fix tile count calculation See merge request !1046
上级
98a0f14e
d111bfc6
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
15 addition
and
9 deletion
+15
-9
mace/utils/count_down_latch.h
mace/utils/count_down_latch.h
+4
-0
mace/utils/count_down_latch_test.cc
mace/utils/count_down_latch_test.cc
+2
-0
mace/utils/thread_pool.cc
mace/utils/thread_pool.cc
+9
-9
未找到文件。
mace/utils/count_down_latch.h
浏览文件 @
cabe5cbe
...
...
@@ -54,6 +54,10 @@ class CountDownLatch {
count_
.
store
(
count
,
std
::
memory_order_release
);
}
int
count
()
const
{
return
count_
;
}
private:
int64_t
spin_timeout_
;
std
::
atomic
<
int
>
count_
;
...
...
mace/utils/count_down_latch_test.cc
浏览文件 @
cabe5cbe
...
...
@@ -39,6 +39,7 @@ TEST_F(CountDownLatchTest, TestWait) {
for
(
int
i
=
0
;
i
<
10
;
++
i
)
{
threads
[
i
].
join
();
}
MACE_CHECK
(
latch
.
count
()
==
0
);
}
TEST_F
(
CountDownLatchTest
,
TestSpinWait
)
{
...
...
@@ -53,6 +54,7 @@ TEST_F(CountDownLatchTest, TestSpinWait) {
for
(
int
i
=
0
;
i
<
10
;
++
i
)
{
threads
[
i
].
join
();
}
MACE_CHECK
(
latch
.
count
()
==
0
);
}
}
// namespace
...
...
mace/utils/thread_pool.cc
浏览文件 @
cabe5cbe
...
...
@@ -134,7 +134,7 @@ ThreadPool::ThreadPool(const size_t thread_count_hint,
default_tile_count_
=
thread_count
;
if
(
cores_to_use
.
size
()
>=
2
&&
cpu_max_freqs
[
0
]
!=
cpu_max_freqs
[
cores_to_use
.
back
()])
{
&&
cpu_max_freqs
[
cores_to_use
[
0
]
]
!=
cpu_max_freqs
[
cores_to_use
.
back
()])
{
default_tile_count_
=
thread_count
*
kTileCountPerThread
;
}
MACE_CHECK
(
default_tile_count_
>
0
,
"default tile count should > 0"
);
...
...
@@ -213,8 +213,8 @@ void ThreadPool::Destroy() {
if
(
threads_
[
i
].
joinable
())
{
threads_
[
i
].
join
();
}
else
{
LOG
(
ERROR
)
<<
"Thread: "
<<
threads_
[
i
].
get_id
()
<<
" not joinable"
<<
std
::
endl
;
VLOG
(
2
)
<<
"Thread: "
<<
threads_
[
i
].
get_id
()
<<
" not joinable"
<<
std
::
endl
;
}
}
}
...
...
@@ -318,7 +318,7 @@ void ThreadPool::Compute1D(const std::function<void(size_t,
}
size_t
step_tile_size
=
step
*
tile_size
;
size_t
tile_count
=
RoundUp
(
items
,
tile_size
);
size_t
tile_count
=
RoundUp
Div
(
items
,
tile_size
);
Run
([
&
](
size_t
tile_idx
)
{
size_t
tile_start
=
start
+
tile_idx
*
step_tile_size
;
size_t
tile_end
=
std
::
min
(
end
,
tile_start
+
step_tile_size
);
...
...
@@ -366,8 +366,8 @@ void ThreadPool::Compute2D(const std::function<void(size_t /* start */,
size_t
step_tile_size0
=
step0
*
tile_size0
;
size_t
step_tile_size1
=
step1
*
tile_size1
;
size_t
tile_count0
=
RoundUp
(
items0
,
tile_size0
);
size_t
tile_count1
=
RoundUp
(
items1
,
tile_size1
);
size_t
tile_count0
=
RoundUp
Div
(
items0
,
tile_size0
);
size_t
tile_count1
=
RoundUp
Div
(
items1
,
tile_size1
);
Run
([
&
](
size_t
tile_idx
)
{
size_t
tile_idx0
=
tile_idx
/
tile_count1
;
...
...
@@ -438,9 +438,9 @@ void ThreadPool::Compute3D(const std::function<void(size_t /* start */,
size_t
step_tile_size0
=
step0
*
tile_size0
;
size_t
step_tile_size1
=
step1
*
tile_size1
;
size_t
step_tile_size2
=
step2
*
tile_size2
;
size_t
tile_count0
=
RoundUp
(
items0
,
tile_size0
);
size_t
tile_count1
=
RoundUp
(
items1
,
tile_size1
);
size_t
tile_count2
=
RoundUp
(
items2
,
tile_size2
);
size_t
tile_count0
=
RoundUp
Div
(
items0
,
tile_size0
);
size_t
tile_count1
=
RoundUp
Div
(
items1
,
tile_size1
);
size_t
tile_count2
=
RoundUp
Div
(
items2
,
tile_size2
);
size_t
tile_count12
=
tile_count1
*
tile_count2
;
Run
([
&
](
size_t
tile_idx
)
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录