You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@singa.apache.org by wa...@apache.org on 2019/03/10 07:18:16 UTC
[incubator-singa] branch master updated (e16209e -> e5f24ce)
This is an automated email from the ASF dual-hosted git repository.
wangwei pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-singa.git.
from e16209e Merge pull request #435 from Zheng130126/master
new 0d0d19a - Integrated MKL dnn for operations below fwd/bwd w/ tests - Conv2D - BatchNorm 4D/2D - Pooling Max/Avg - Updated CMake for flag: USE_MKLDNN
new 4b33432 fixed batchnorm moving avg, turn off option USE_MKLDNN by default
new 717d72a mkldnn-tested-python-api
new b96293c updated docs installation for MKL-DNN
new c24c688 added cmake find mkldnn lib
new e6a7e35 type
new e5f24ce Merge pull request #431 from dcslin/SINGA-MKLDNN-Integration
The 1094 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
CMakeLists.txt | 1 +
cmake/Dependencies.cmake | 7 +
cmake/Templates/singa_config.h.in | 2 +
doc/en/docs/installation.md | 16 ++
include/singa/core/common.h | 8 +
include/singa/core/device.h | 6 +-
include/singa/utils/mkldnn_utils.h | 36 +++++
src/api/config.i.in | 1 +
src/api/model_operation.i | 31 ++++
src/core/device/cpp_cpu.cc | 9 ++
src/model/operation/batchnorm.cc | 189 +++++++++++++++++++++++
src/model/operation/batchnorm.h | 47 +++++-
src/model/operation/convolution.cc | 157 ++++++++++++++++++-
src/model/operation/convolution.h | 24 +++
src/model/operation/pooling.cc | 117 ++++++++++++++
src/model/operation/pooling.h | 29 ++++
test/python/test_mkldnn.py | 172 +++++++++++++++++++++
test/singa/test_operation_batchnorm.cc | 252 +++++++++++++++++++++++++++++++
test/singa/test_operation_convolution.cc | 171 +++++++++++++++++++++
test/singa/test_operation_pooling.cc | 183 ++++++++++++++++++++++
20 files changed, 1451 insertions(+), 7 deletions(-)
create mode 100644 include/singa/utils/mkldnn_utils.h
create mode 100755 test/python/test_mkldnn.py
create mode 100644 test/singa/test_operation_batchnorm.cc
create mode 100644 test/singa/test_operation_convolution.cc
create mode 100644 test/singa/test_operation_pooling.cc