未验证 提交 1dce992d 编写于 作者: B BUG1989 提交者: GitHub

Fix, the affinity of cpu is error (#687)

上级 1a54989a
......@@ -516,7 +516,7 @@ int prerun_graph_multithread(graph_t graph, struct options option)
mask = get_cpu_cluster_mask(option.cluster);
}
int count = get_mask_count(mask);
int count = get_cpu_mask_count(mask);
if (0 < option.num_thread && count > option.num_thread)
{
count = option.num_thread;
......@@ -537,7 +537,7 @@ int prerun_graph_multithread(graph_t graph, struct options option)
opt->num_thread = count;
opt->cluster = TENGINE_CLUSTER_BIG;
opt->precision = precision;
opt->affinity = mask;
opt->affinity = option.affinity;
struct scheduler* scheduler = ctx->scheduler;
ret = scheduler->prerun(scheduler, ir_graph);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册