Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
Xiaomi
Mace
提交
ab81d04a
Mace
项目概览
Xiaomi
/
Mace
通知
106
Star
40
Fork
27
代码
文件
提交
分支
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,发现更多精彩内容 >>
提交
ab81d04a
编写于
7月 13, 2018
作者:
李
李滨
浏览文件
操作
浏览文件
下载
差异文件
Merge branch 'fix-build-standalone' into 'master'
Fix bug: build x86_64 which need opencl. See merge request !659
上级
3713852b
f60de41d
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
18 addition
and
0 deletion
+18
-0
.gitlab-ci.yml
.gitlab-ci.yml
+6
-0
mace/libmace/mace_runtime.cc
mace/libmace/mace_runtime.cc
+12
-0
未找到文件。
.gitlab-ci.yml
浏览文件 @
ab81d04a
...
...
@@ -3,6 +3,7 @@ stages:
-
pycodestyle
-
docs
-
platform_compatible_tests
-
build_libraries
-
ndk_versions_compatible_tests
-
ops_test
-
api_test
...
...
@@ -72,6 +73,11 @@ platform_compatible_tests:
script
:
-
bazel build mace/core:core
build_libraries
:
stage
:
build_libraries
script
:
-
bash tools/build-standalone-lib.sh
ndk_versions_compatible_tests
:
stage
:
ndk_versions_compatible_tests
script
:
...
...
mace/libmace/mace_runtime.cc
浏览文件 @
ab81d04a
...
...
@@ -12,8 +12,11 @@
// See the License for the specific language governing permissions and
// limitations under the License.
#include "mace/core/macros.h"
#include "mace/core/runtime/cpu/cpu_runtime.h"
#ifdef MACE_ENABLE_OPENCL
#include "mace/core/runtime/opencl/opencl_runtime.h"
#endif // MACE_ENABLE_OPENCL
#include "mace/public/mace_runtime.h"
#include "mace/utils/logging.h"
...
...
@@ -28,7 +31,11 @@ void SetKVStorageFactory(std::shared_ptr<KVStorageFactory> storage_factory) {
// Set OpenCL Compiled Binary paths, just call once. (Not thread-safe)
void
SetOpenCLBinaryPaths
(
const
std
::
vector
<
std
::
string
>
&
paths
)
{
#ifdef MACE_ENABLE_OPENCL
OpenCLRuntime
::
ConfigureOpenCLBinaryPath
(
paths
);
#else
MACE_UNUSED
(
paths
);
#endif // MACE_ENABLE_OPENCL
}
extern
std
::
string
kOpenCLParameterPath
;
...
...
@@ -38,9 +45,14 @@ void SetOpenCLParameterPath(const std::string &path) {
}
void
SetGPUHints
(
GPUPerfHint
gpu_perf_hint
,
GPUPriorityHint
gpu_priority_hint
)
{
#ifdef MACE_ENABLE_OPENCL
VLOG
(
1
)
<<
"Set GPU configurations, gpu_perf_hint: "
<<
gpu_perf_hint
<<
", gpu_priority_hint: "
<<
gpu_priority_hint
;
OpenCLRuntime
::
Configure
(
gpu_perf_hint
,
gpu_priority_hint
);
#else
MACE_UNUSED
(
gpu_perf_hint
);
MACE_UNUSED
(
gpu_priority_hint
);
#endif // MACE_ENABLE_OPENCL
}
MaceStatus
SetOpenMPThreadPolicy
(
int
num_threads_hint
,
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录