Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

[MKL-DNN] Add mxnet mkldnn cmake tutorial #16688

Merged
merged 3 commits into from
Nov 1, 2019
Merged

[MKL-DNN] Add mxnet mkldnn cmake tutorial #16688

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
fc245dc
add mxnet mkldnn cmake instruction
xinyu-intel Nov 1, 2019
3fa3074
imporve doc
xinyu-intel Nov 1, 2019
72acfb3
OMP->OpenMP
xinyu-intel Nov 1, 2019
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 14 additions & 1 deletion docs/python_docs/python/tutorials/performance/backend/mkldnn/mkldnn_readme.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,11 +57,24 @@ cd incubator-mxnet

### Build MXNet with MKL-DNN

To achieve better performance, the Intel OpenMP and llvm OpenMP are recommended as below instruction. Otherwise, default GNU OpenMP will be used and you may get the sub-optimal performance. If you don't have the full [MKL](https://software.intel.com/en-us/intel-mkl) library installation, you might use OpenBLAS as the blas library, by setting USE_BLAS=openblas.

```
# build with llvm OpenMP and Intel MKL/openblas
mkdir build && cd build
cmake -DUSE_CUDA=OFF -DUSE_MKL_IF_AVAILABLE=ON -DUSE_MKLDNN=ON -DUSE_OPENMP=ON -DUSE_OPENCV=ON ..
make -j $(nproc)
```

```
# build with Intel MKL and Intel OpenMP
make -j $(nproc) USE_OPENCV=1 USE_MKLDNN=1 USE_BLAS=mkl USE_INTEL_PATH=/opt/intel
```

If you don't have the full [MKL](https://software.intel.com/en-us/intel-mkl) library installation, you might use OpenBLAS as the blas library, by setting USE_BLAS=openblas.
```
# build with openblas and GNU OpenMP(sub-optimal performance)
make -j $(nproc) USE_OPENCV=1 USE_MKLDNN=1 USE_BLAS=openblas
```

<h2 id="2">MacOS</h2>

Expand Down