Skip to content

Commit 74de8c0

Browse files
committed
make docker can configure with_mkldnn and with_mklml
1 parent c169669 commit 74de8c0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

paddle/scripts/docker/build.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@ Configuring cmake in /paddle/build ...
3030
-DCMAKE_BUILD_TYPE=Release
3131
-DWITH_DOC=OFF
3232
-DWITH_GPU=${WITH_GPU:-OFF}
33+
-DWITH_MKLDNN=${WITH_MKLDNN:-ON}
34+
-DWITH_MKLML=${WITH_MKLML:-ON}
3335
-DWITH_AVX=${WITH_AVX:-OFF}
3436
-DWITH_GOLANG=${WITH_GOLANG:-ON}
3537
-DWITH_SWIG_PY=ON
@@ -50,6 +52,8 @@ cmake .. \
5052
-DCMAKE_BUILD_TYPE=Release \
5153
-DWITH_DOC=OFF \
5254
-DWITH_GPU=${WITH_GPU:-OFF} \
55+
-DWITH_MKLDNN=${WITH_MKLDNN:-ON} \
56+
-DWITH_MKLML=${WITH_MKLML:-ON} \
5357
-DWITH_AVX=${WITH_AVX:-OFF} \
5458
-DWITH_GOLANG=${WITH_GOLANG:-ON} \
5559
-DWITH_SWIG_PY=${WITH_SWIG_PY:-ON} \

0 commit comments

Comments
 (0)