Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
as350144
Mace
提交
ab1f2de8
Mace
项目概览
as350144
/
Mace
与 Fork 源项目一致
Fork自
Xiaomi / Mace
通知
2
Star
1
Fork
1
代码
文件
提交
分支
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,发现更多精彩内容 >>
提交
ab1f2de8
编写于
10月 10, 2017
作者:
L
Liangliang He
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Add OpenCL 1.2 shared library loader and wrapper
上级
4943f2aa
变更
5
展开全部
隐藏空白更改
内联
并排
Showing
5 changed file
with
674 addition
and
38 deletion
+674
-38
WORKSPACE
WORKSPACE
+27
-24
mace/core/BUILD
mace/core/BUILD
+6
-2
mace/core/platform/opencl/opencl_wrapper.cc
mace/core/platform/opencl/opencl_wrapper.cc
+615
-0
mace/core/platform/opencl/opencl_wrapper.h
mace/core/platform/opencl/opencl_wrapper.h
+14
-0
mace/third_party/opencl-headers.BUILD
mace/third_party/opencl-headers.BUILD
+12
-12
未找到文件。
WORKSPACE
浏览文件 @
ab1f2de8
workspace
(
name
=
"mace"
)
workspace
(
name
=
"mace"
)
# proto_library rules implicitly depend on @com_google_protobuf//:protoc,
# which is the proto-compiler.
# This statement defines the @com_google_protobuf repo.
http_archive
(
http_archive
(
name
=
"
org_tensorflow
"
,
name
=
"
com_google_protobuf
"
,
urls
=
[
"http://v9.git.n.xiaomi.com/deep-learning/
tensorflow/repository/archive.zip?ref=v1.3.0
"
],
urls
=
[
"http://v9.git.n.xiaomi.com/deep-learning/
protobuf/repository/archive.zip?ref=c7457ef65a7a8584b1e3bd396c401ccf8e275ffa
"
],
strip_prefix
=
"
tensorflow-v1.3.0-9e76bf324f6bac63137a02bb6e6ec9120703ea9b
"
,
strip_prefix
=
"
protobuf-c7457ef65a7a8584b1e3bd396c401ccf8e275ffa-c7457ef65a7a8584b1e3bd396c401ccf8e275ffa
"
,
sha256
=
"
97049d3a59a77858e12c55422bd129261b14e869a91aebcdcc39439393c00dc7
"
,
sha256
=
"
0a54cae83b77f4b54b7db4eaebadd81fbe91655e84a1ef3f6d29116d75f3a45f
"
,
)
)
#
TensorFlow depends on "io_bazel_rules_closure" so we need this here.
#
cc_proto_library rules implicitly depend on @com_google_protobuf_cc//:cc_toolchain,
#
Needs to be kept in sync with the same target in TensorFlow's WORKSPACE file
.
#
which is the C++ proto runtime (base classes and common utilities)
.
http_archive
(
http_archive
(
name
=
"io_bazel_rules_closure"
,
name
=
"com_google_protobuf_cc"
,
sha256
=
"60fc6977908f999b23ca65698c2bb70213403824a84f7904310b6000d78be9ce"
,
urls
=
[
"http://v9.git.n.xiaomi.com/deep-learning/protobuf/repository/archive.zip?ref=c7457ef65a7a8584b1e3bd396c401ccf8e275ffa"
],
strip_prefix
=
"rules_closure-5ca1dab6df9ad02050f7ba4e816407f88690cf7d"
,
strip_prefix
=
"protobuf-c7457ef65a7a8584b1e3bd396c401ccf8e275ffa-c7457ef65a7a8584b1e3bd396c401ccf8e275ffa"
,
urls
=
[
sha256
=
"0a54cae83b77f4b54b7db4eaebadd81fbe91655e84a1ef3f6d29116d75f3a45f"
,
"http://bazel-mirror.storage.googleapis.com/github.com/bazelbuild/rules_closure/archive/5ca1dab6df9ad02050f7ba4e816407f88690cf7d.tar.gz"
,
# 2017-02-03
"https://github.com/bazelbuild/rules_closure/archive/5ca1dab6df9ad02050f7ba4e816407f88690cf7d.tar.gz"
,
],
)
)
new_http_archive
(
new_http_archive
(
...
@@ -27,16 +27,19 @@ new_http_archive(
...
@@ -27,16 +27,19 @@ new_http_archive(
build_file
=
"mace/third_party/gtest.BUILD"
,
build_file
=
"mace/third_party/gtest.BUILD"
,
)
)
# Import all of the tensorflow dependencies.
load
(
'@org_tensorflow//tensorflow:workspace.bzl'
,
'tf_workspace'
)
tf_workspace
(
tf_repo_name
=
"org_tensorflow"
)
new_http_archive
(
new_http_archive
(
name
=
"ncnn"
,
name
=
"six_archive"
,
urls
=
[
"http://v9.git.n.xiaomi.com/deep-learning/ncnn/repository/archive.zip?ref=bazel-fix"
],
urls
=
[
strip_prefix
=
"ncnn-bazel-fix-ce5e416164545e1ab37fe3544502624f605ca234/src"
,
"http://mirror.bazel.build/pypi.python.org/packages/source/s/six/six-1.10.0.tar.gz"
,
sha256
=
"e6d76356179bcdbb988279f0b42ab050c8af55970e1ad767787ad21d5b7aad51"
,
"https://pypi.python.org/packages/source/s/six/six-1.10.0.tar.gz"
,
build_file
=
"mace/third_party/ncnn.BUILD"
,
],
sha256
=
"105f8d68616f8248e24bf0e9372ef04d3cc10104f1980f54d57b2ce73a5ad56a"
,
strip_prefix
=
"six-1.10.0"
,
build_file
=
"mace/third_party/six.BUILD"
,
)
bind
(
name
=
"six"
,
actual
=
"@six_archive//:six"
,
)
)
new_http_archive
(
new_http_archive
(
...
@@ -50,6 +53,6 @@ new_http_archive(
...
@@ -50,6 +53,6 @@ new_http_archive(
# Set up Android NDK
# Set up Android NDK
android_ndk_repository
(
android_ndk_repository
(
name
=
"androidndk"
,
name
=
"androidndk"
,
# Android
4
.0
# Android
5
.0
api_level
=
14
api_level
=
21
)
)
mace/core/BUILD
浏览文件 @
ab1f2de8
...
@@ -12,11 +12,15 @@ load("//mace:mace.bzl", "if_android")
...
@@ -12,11 +12,15 @@ load("//mace:mace.bzl", "if_android")
cc_library
(
cc_library
(
name
=
"opencl_runtime"
,
name
=
"opencl_runtime"
,
srcs
=
glob
([
srcs
=
glob
([
"platform/opencl/*.hpp"
,
"platform/opencl/cl.hpp"
,
"platform/opencl/cl2.hpp"
,
"platform/opencl/opencl_wrapper.h"
,
"platform/opencl/opencl_wrapper.cc"
,
]),
]),
copts
=
[
"-std=c++11"
],
copts
=
[
"-std=c++11"
],
deps
=
[
deps
=
[
"@opencl_headers//:opencl20_headers"
,
"@opencl_headers//:opencl12_headers"
,
"core"
,
],
],
)
)
...
...
mace/core/platform/opencl/opencl_wrapper.cc
0 → 100644
浏览文件 @
ab1f2de8
此差异已折叠。
点击以展开。
mace/core/platform/opencl/opencl_wrapper.h
0 → 100644
浏览文件 @
ab1f2de8
//
// Copyright (c) 2017 XiaoMi All rights reserved.
//
#include "CL/opencl.h"
#include "mace/core/logging.h"
#include <dlfcn.h>
#include <mutex>
namespace
mace
{
bool
OpenCLSupported
();
}
// namespace mace
mace/third_party/opencl-headers.BUILD
浏览文件 @
ab1f2de8
cc_
inc_
library(
cc_library(
name = "opencl10_headers",
name = "opencl10_headers",
hdrs = glob([
hdrs = glob([
"opencl10/CL/*.h",
"opencl10/CL/*.h",
]),
]),
prefix = "opencl10",
strip_include_
prefix = "opencl10",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
cc_
inc_
library(
cc_library(
name = "opencl11_headers",
name = "opencl11_headers",
hdrs = glob([
hdrs = glob([
"opencl11/CL/*.h",
"opencl11/CL/*.h",
]),
]),
prefix = "opencli
11",
strip_include_prefix = "opencl
11",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
cc_
inc_
library(
cc_library(
name = "opencl12_headers",
name = "opencl12_headers",
hdrs = glob([
hdrs = glob([
"opencl12/CL/*.h",
"opencl12/CL/*.h",
]),
]),
prefix = "opencl12",
strip_include_
prefix = "opencl12",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
cc_
inc_
library(
cc_library(
name = "opencl20_headers",
name = "opencl20_headers",
hdrs = glob([
hdrs = glob([
"opencl20/CL/*.h",
"opencl20/CL/*.h",
]),
]),
prefix = "opencl20",
strip_include_
prefix = "opencl20",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
cc_
inc_
library(
cc_library(
name = "opencl21_headers",
name = "opencl21_headers",
hdrs = glob([
hdrs = glob([
"opencl21/CL/*.h",
"opencl21/CL/*.h",
]),
]),
prefix = "opencl21",
strip_include_
prefix = "opencl21",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
cc_
inc_
library(
cc_library(
name = "opencl22_headers",
name = "opencl22_headers",
hdrs = glob([
hdrs = glob([
"opencl22/CL/*.h",
"opencl22/CL/*.h",
]),
]),
prefix = "opencl22",
strip_include_
prefix = "opencl22",
visibility = ["//visibility:public"],
visibility = ["//visibility:public"],
)
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录