You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by la...@apache.org on 2021/03/24 14:17:31 UTC
[incubator-mxnet] branch master updated: Change *_mkldnn* test and
build scenarios names to *_onednn* (#20034)
This is an automated email from the ASF dual-hosted git repository.
lausen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new c7a8ccc Change *_mkldnn* test and build scenarios names to *_onednn* (#20034)
c7a8ccc is described below
commit c7a8ccc7220d0d710e5274d075c4ce4f55c81c37
Author: bartekkuncer <ba...@intel.com>
AuthorDate: Wed Mar 24 15:15:32 2021 +0100
Change *_mkldnn* test and build scenarios names to *_onednn* (#20034)
---
ci/build_windows.py | 12 ++--
ci/dev_menu.py | 8 +--
ci/docker/runtime_functions.sh | 18 ++---
ci/jenkins/Jenkins_steps.groovy | 130 +++++++++++++++++------------------
ci/jenkins/Jenkinsfile_centos_cpu | 2 +-
ci/jenkins/Jenkinsfile_clang | 4 +-
ci/jenkins/Jenkinsfile_unix_cpu | 8 +--
ci/jenkins/Jenkinsfile_unix_gpu | 8 +--
ci/jenkins/Jenkinsfile_windows_cpu | 4 +-
ci/jenkins/Jenkinsfile_windows_gpu | 4 +-
tests/nightly/JenkinsfileForBinaries | 4 +-
11 files changed, 101 insertions(+), 101 deletions(-)
diff --git a/ci/build_windows.py b/ci/build_windows.py
index 2035e4c..fe2a03b 100755
--- a/ci/build_windows.py
+++ b/ci/build_windows.py
@@ -48,11 +48,11 @@ KNOWN_VCVARS = {
class BuildFlavour(Enum):
WIN_CPU = 'WIN_CPU'
- WIN_CPU_MKLDNN = 'WIN_CPU_MKLDNN'
- WIN_CPU_MKLDNN_MKL = 'WIN_CPU_MKLDNN_MKL'
+ WIN_CPU_ONEDNN = 'WIN_CPU_ONEDNN'
+ WIN_CPU_ONEDNN_MKL = 'WIN_CPU_ONEDNN_MKL'
WIN_CPU_MKL = 'WIN_CPU_MKL'
WIN_GPU = 'WIN_GPU'
- WIN_GPU_MKLDNN = 'WIN_GPU_MKLDNN'
+ WIN_GPU_ONEDNN = 'WIN_GPU_ONEDNN'
CMAKE_FLAGS = {
@@ -69,7 +69,7 @@ CMAKE_FLAGS = {
'-DBUILD_CPP_EXAMPLES=ON '
'-DCMAKE_BUILD_TYPE=Release')
- , 'WIN_CPU_MKLDNN': (
+ , 'WIN_CPU_ONEDNN': (
'-DCMAKE_C_COMPILER=cl '
'-DCMAKE_CXX_COMPILER=cl '
'-DUSE_CUDA=OFF '
@@ -82,7 +82,7 @@ CMAKE_FLAGS = {
'-DUSE_ONEDNN=ON '
'-DCMAKE_BUILD_TYPE=Release')
- , 'WIN_CPU_MKLDNN_MKL': (
+ , 'WIN_CPU_ONEDNN_MKL': (
'-DCMAKE_C_COMPILER=cl '
'-DCMAKE_CXX_COMPILER=cl '
'-DUSE_CUDA=OFF '
@@ -121,7 +121,7 @@ CMAKE_FLAGS = {
'-DMXNET_CUDA_ARCH="5.2" '
'-DCMAKE_BUILD_TYPE=Release')
- , 'WIN_GPU_MKLDNN': (
+ , 'WIN_GPU_ONEDNN': (
'-DCMAKE_C_COMPILER=cl '
'-DCMAKE_CXX_COMPILER=cl '
'-DUSE_CUDA=ON '
diff --git a/ci/dev_menu.py b/ci/dev_menu.py
index 4e996bf..a21129c 100644
--- a/ci/dev_menu.py
+++ b/ci/dev_menu.py
@@ -141,14 +141,14 @@ COMMANDS = OrderedDict([
"ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu",
"ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python3_gpu",
]),
- ('[Docker] Python3 GPU+MKLDNN unittests',
+ ('[Docker] Python3 GPU+ONEDNN unittests',
[
- "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu_cmake_mkldnn",
+ "ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh build_ubuntu_gpu_onednn",
"ci/build.py --nvidiadocker --platform ubuntu_gpu /work/runtime_functions.sh unittest_ubuntu_python3_gpu",
]),
- ('[Docker] Python3 CPU Intel MKLDNN unittests',
+ ('[Docker] Python3 CPU Intel ONEDNN unittests',
[
- "ci/build.py --platform ubuntu_cpu /work/runtime_functions.sh build_ubuntu_cpu_mkldnn",
+ "ci/build.py --platform ubuntu_cpu /work/runtime_functions.sh build_ubuntu_cpu_onednn",
"ci/build.py --platform ubuntu_cpu /work/runtime_functions.sh unittest_ubuntu_python3_cpu",
]),
('[Docker] Python3 ARMv7 unittests (QEMU)',
diff --git a/ci/docker/runtime_functions.sh b/ci/docker/runtime_functions.sh
index 078b4ff..e53a57b 100755
--- a/ci/docker/runtime_functions.sh
+++ b/ci/docker/runtime_functions.sh
@@ -273,7 +273,7 @@ build_centos7_cpu() {
ninja
}
-build_centos7_mkldnn() {
+build_centos7_onednn() {
set -ex
cd /work/build
source /opt/rh/devtoolset-7/enable
@@ -483,7 +483,7 @@ build_ubuntu_cpu_clang_tidy() {
ninja
}
-build_ubuntu_cpu_clang6_mkldnn() {
+build_ubuntu_cpu_clang6_onednn() {
set -ex
cd /work/build
export OpenBLAS_HOME=/usr/local/openblas-clang/
@@ -496,7 +496,7 @@ build_ubuntu_cpu_clang6_mkldnn() {
ninja
}
-build_ubuntu_cpu_clang100_mkldnn() {
+build_ubuntu_cpu_clang100_onednn() {
set -ex
cd /work/build
export OpenBLAS_HOME=/usr/local/openblas-clang/
@@ -508,7 +508,7 @@ build_ubuntu_cpu_clang100_mkldnn() {
ninja
}
-build_ubuntu_cpu_mkldnn() {
+build_ubuntu_cpu_onednn() {
set -ex
cd /work/build
CC=gcc-7 CXX=g++-7 cmake \
@@ -523,7 +523,7 @@ build_ubuntu_cpu_mkldnn() {
ninja
}
-build_ubuntu_cpu_mkldnn_mkl() {
+build_ubuntu_cpu_onednn_mkl() {
set -ex
cd /work/build
CC=gcc-7 CXX=g++-7 cmake \
@@ -593,7 +593,7 @@ build_ubuntu_gpu_tensorrt() {
ninja
}
-build_ubuntu_gpu_mkldnn() {
+build_ubuntu_gpu_onednn() {
set -ex
cd /work/build
CC=gcc-7 CXX=g++-7 cmake \
@@ -607,7 +607,7 @@ build_ubuntu_gpu_mkldnn() {
ninja
}
-build_ubuntu_gpu_mkldnn_nocudnn() {
+build_ubuntu_gpu_onednn_nocudnn() {
set -ex
cd /work/build
CC=gcc-7 CXX=g++-7 cmake \
@@ -775,7 +775,7 @@ unittest_ubuntu_python3_cpu() {
pytest -m 'serial' --durations=50 --cov-report xml:tests_unittest.xml --cov-append --verbose tests/python/unittest
}
-unittest_ubuntu_python3_cpu_mkldnn() {
+unittest_ubuntu_python3_cpu_onednn() {
set -ex
export PYTHONPATH=./python/
export MXNET_ONEDNN_DEBUG=0 # Ignored if not present
@@ -1291,7 +1291,7 @@ build_static_libmxnet() {
# Tests CD PyPI packaging in CI
ci_package_pypi() {
set -ex
- # copies mkldnn header files to 3rdparty/onednn/include/oneapi/dnnl/ as in CD
+ # copies onednn header files to 3rdparty/onednn/include/oneapi/dnnl/ as in CD
mkdir -p 3rdparty/onednn/include/oneapi/dnnl
cp include/onednn/oneapi/dnnl/dnnl_version.h 3rdparty/onednn/include/oneapi/dnnl/.
cp include/onednn/oneapi/dnnl/dnnl_config.h 3rdparty/onednn/include/oneapi/dnnl/.
diff --git a/ci/jenkins/Jenkins_steps.groovy b/ci/jenkins/Jenkins_steps.groovy
index 7b0ee7e..ae37862 100644
--- a/ci/jenkins/Jenkins_steps.groovy
+++ b/ci/jenkins/Jenkins_steps.groovy
@@ -48,9 +48,9 @@ def python3_ut(docker_container_name) {
}
}
-def python3_ut_mkldnn(docker_container_name) {
+def python3_ut_onednn(docker_container_name) {
timeout(time: max_time, unit: 'MINUTES') {
- utils.docker_run(docker_container_name, 'unittest_ubuntu_python3_cpu_mkldnn', false)
+ utils.docker_run(docker_container_name, 'unittest_ubuntu_python3_cpu_onednn', false)
}
}
@@ -161,13 +161,13 @@ def compile_unix_mkl_cpu(lib_name) {
}]
}
-def compile_unix_mkldnn_cpu(lib_name) {
- return ['CPU: MKLDNN': {
+def compile_unix_onednn_cpu(lib_name) {
+ return ['CPU: ONEDNN': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-mkldnn-cpu') {
+ ws('workspace/build-onednn-cpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_mkldnn', false)
+ utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_onednn', false)
utils.pack_lib(lib_name, mx_onednn_lib, true)
}
}
@@ -175,13 +175,13 @@ def compile_unix_mkldnn_cpu(lib_name) {
}]
}
-def compile_unix_mkldnn_mkl_cpu(lib_name) {
- return ['CPU: MKLDNN_MKL': {
+def compile_unix_onednn_mkl_cpu(lib_name) {
+ return ['CPU: ONEDNN_MKL': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-mkldnn-cpu') {
+ ws('workspace/build-onednn-cpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_mkldnn_mkl', false)
+ utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_onednn_mkl', false)
utils.pack_lib(lib_name, mx_onednn_lib, false)
}
}
@@ -189,13 +189,13 @@ def compile_unix_mkldnn_mkl_cpu(lib_name) {
}]
}
-def compile_unix_mkldnn_gpu(lib_name) {
- return ['GPU: MKLDNN': {
+def compile_unix_onednn_gpu(lib_name) {
+ return ['GPU: ONEDNN': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-mkldnn-gpu') {
+ ws('workspace/build-onednn-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_mkldnn', false)
+ utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_onednn', false)
utils.pack_lib(lib_name, mx_onednn_lib)
}
}
@@ -203,13 +203,13 @@ def compile_unix_mkldnn_gpu(lib_name) {
}]
}
-def compile_unix_mkldnn_nocudnn_gpu(lib_name) {
- return ['GPU: MKLDNN_CUDNNOFF': {
+def compile_unix_onednn_nocudnn_gpu(lib_name) {
+ return ['GPU: ONEDNN_CUDNNOFF': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-mkldnn-gpu-nocudnn') {
+ ws('workspace/build-onednn-gpu-nocudnn') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_mkldnn_nocudnn', false)
+ utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_onednn_nocudnn', false)
utils.pack_lib(lib_name, mx_onednn_lib)
}
}
@@ -273,13 +273,13 @@ def compile_centos7_cpu(lib_name) {
}]
}
-def compile_centos7_cpu_mkldnn() {
- return ['CPU: CentOS 7 MKLDNN': {
+def compile_centos7_cpu_onednn() {
+ return ['CPU: CentOS 7 ONEDNN': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-centos7-mkldnn') {
+ ws('workspace/build-centos7-onednn') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('centos7_cpu', 'build_centos7_mkldnn', false)
+ utils.docker_run('centos7_cpu', 'build_centos7_onednn', false)
}
}
}
@@ -340,13 +340,13 @@ def compile_unix_clang_tidy_cpu() {
}]
}
-def compile_unix_clang_6_mkldnn_cpu() {
- return ['CPU: Clang 6 MKLDNN': {
+def compile_unix_clang_6_onednn_cpu() {
+ return ['CPU: Clang 6 ONEDNN': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-cpu-mkldnn-clang6') {
+ ws('workspace/build-cpu-onednn-clang6') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_clang6_mkldnn', false)
+ utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_clang6_onednn', false)
}
}
}
@@ -354,13 +354,13 @@ def compile_unix_clang_6_mkldnn_cpu() {
}
// TODO(leezu) delete once DUSE_DIST_KVSTORE=ON builds in -WError build
-def compile_unix_clang_10_mkldnn_cpu() {
- return ['CPU: Clang 10 MKLDNN': {
+def compile_unix_clang_10_onednn_cpu() {
+ return ['CPU: Clang 10 ONEDNN': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-cpu-mkldnn-clang100') {
+ ws('workspace/build-cpu-onednn-clang100') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
- utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_clang100_mkldnn', false)
+ utils.docker_run('ubuntu_cpu', 'build_ubuntu_cpu_clang100_onednn', false)
}
}
}
@@ -518,13 +518,13 @@ def compile_windows_cpu(lib_name) {
}]
}
-def compile_windows_cpu_mkldnn(lib_name) {
- return ['Build CPU MKLDNN windows':{
+def compile_windows_cpu_onednn(lib_name) {
+ return ['Build CPU ONEDNN windows':{
node(NODE_WINDOWS_CPU) {
- ws('workspace/build-cpu-mkldnn') {
+ ws('workspace/build-cpu-onednn') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git_win()
- powershell 'py -3 ci/build_windows.py -f WIN_CPU_MKLDNN'
+ powershell 'py -3 ci/build_windows.py -f WIN_CPU_ONEDNN'
stash includes: 'windows_package.7z', name: lib_name
}
}
@@ -532,13 +532,13 @@ def compile_windows_cpu_mkldnn(lib_name) {
}]
}
-def compile_windows_cpu_mkldnn_mkl(lib_name) {
- return ['Build CPU MKLDNN MKL windows':{
+def compile_windows_cpu_onednn_mkl(lib_name) {
+ return ['Build CPU ONEDNN MKL windows':{
node(NODE_WINDOWS_CPU) {
- ws('workspace/build-cpu-mkldnn-mkl') {
+ ws('workspace/build-cpu-onednn-mkl') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git_win()
- powershell 'py -3 ci/build_windows.py -f WIN_CPU_MKLDNN_MKL'
+ powershell 'py -3 ci/build_windows.py -f WIN_CPU_ONEDNN_MKL'
stash includes: 'windows_package.7z', name: lib_name
}
}
@@ -574,13 +574,13 @@ def compile_windows_gpu(lib_name) {
}]
}
-def compile_windows_gpu_mkldnn(lib_name) {
- return ['Build GPU MKLDNN windows':{
+def compile_windows_gpu_onednn(lib_name) {
+ return ['Build GPU ONEDNN windows':{
node(NODE_WINDOWS_CPU) {
ws('workspace/build-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git_win()
- powershell 'py -3 ci/build_windows.py -f WIN_GPU_MKLDNN'
+ powershell 'py -3 ci/build_windows.py -f WIN_GPU_ONEDNN'
stash includes: 'windows_package.7z', name: lib_name
}
}
@@ -724,66 +724,66 @@ def test_unix_python3_cpu_no_tvm_op(lib_name) {
}]
}
-def test_unix_python3_mkldnn_cpu(lib_name) {
- return ['Python3: MKLDNN-CPU': {
+def test_unix_python3_onednn_cpu(lib_name) {
+ return ['Python3: ONEDNN-CPU': {
node(NODE_LINUX_CPU) {
- ws('workspace/ut-python3-mkldnn-cpu') {
+ ws('workspace/ut-python3-onednn-cpu') {
try {
utils.unpack_and_init(lib_name, mx_onednn_lib, true)
- python3_ut_mkldnn('ubuntu_cpu')
+ python3_ut_onednn('ubuntu_cpu')
utils.publish_test_coverage()
} finally {
- utils.collect_test_results_unix('tests_unittest.xml', 'tests_python3_mkldnn_cpu_unittest.xml')
- utils.collect_test_results_unix('tests_mkl.xml', 'tests_python3_mkldnn_cpu_mkl.xml')
+ utils.collect_test_results_unix('tests_unittest.xml', 'tests_python3_onednn_cpu_unittest.xml')
+ utils.collect_test_results_unix('tests_mkl.xml', 'tests_python3_onednn_cpu_mkl.xml')
}
}
}
}]
}
-def test_unix_python3_mkldnn_mkl_cpu(lib_name) {
- return ['Python3: MKLDNN-MKL-CPU': {
+def test_unix_python3_onednn_mkl_cpu(lib_name) {
+ return ['Python3: ONEDNN-MKL-CPU': {
node(NODE_LINUX_CPU) {
- ws('workspace/ut-python3-mkldnn-mkl-cpu') {
+ ws('workspace/ut-python3-onednn-mkl-cpu') {
try {
utils.unpack_and_init(lib_name, mx_lib)
- python3_ut_mkldnn('ubuntu_cpu')
+ python3_ut_onednn('ubuntu_cpu')
utils.publish_test_coverage()
} finally {
- utils.collect_test_results_unix('tests_unittest.xml', 'tests_python3_mkldnn_cpu_unittest.xml')
- utils.collect_test_results_unix('tests_mkl.xml', 'tests_python3_mkldnn_cpu_mkl.xml')
+ utils.collect_test_results_unix('tests_unittest.xml', 'tests_python3_onednn_cpu_unittest.xml')
+ utils.collect_test_results_unix('tests_mkl.xml', 'tests_python3_onednn_cpu_mkl.xml')
}
}
}
}]
}
-def test_unix_python3_mkldnn_gpu(lib_name) {
- return ['Python3: MKLDNN-GPU': {
+def test_unix_python3_onednn_gpu(lib_name) {
+ return ['Python3: ONEDNN-GPU': {
node(NODE_LINUX_GPU_G4) {
- ws('workspace/ut-python3-mkldnn-gpu') {
+ ws('workspace/ut-python3-onednn-gpu') {
try {
utils.unpack_and_init(lib_name, mx_onednn_lib)
python3_gpu_ut('ubuntu_gpu_cu111')
utils.publish_test_coverage()
} finally {
- utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_mkldnn_gpu.xml')
+ utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_onednn_gpu.xml')
}
}
}
}]
}
-def test_unix_python3_mkldnn_nocudnn_gpu(lib_name) {
- return ['Python3: MKLDNN-GPU-NOCUDNN': {
+def test_unix_python3_onednn_nocudnn_gpu(lib_name) {
+ return ['Python3: ONEDNN-GPU-NOCUDNN': {
node(NODE_LINUX_GPU_G4) {
- ws('workspace/ut-python3-mkldnn-gpu-nocudnn') {
+ ws('workspace/ut-python3-onednn-gpu-nocudnn') {
try {
utils.unpack_and_init(lib_name, mx_onednn_lib)
python3_gpu_ut_nocudnn('ubuntu_gpu_cu111')
utils.publish_test_coverage()
} finally {
- utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_mkldnn_gpu_nocudnn.xml')
+ utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_onednn_gpu_nocudnn.xml')
}
}
}
@@ -954,8 +954,8 @@ def test_windows_python3_gpu(lib_name) {
}]
}
-def test_windows_python3_gpu_mkldnn(lib_name) {
- return ['Python 3: MKLDNN-GPU Win':{
+def test_windows_python3_gpu_onednn(lib_name) {
+ return ['Python 3: ONEDNN-GPU Win':{
node(NODE_WINDOWS_GPU) {
timeout(time: max_time, unit: 'MINUTES') {
ws('workspace/ut-python-gpu') {
@@ -964,8 +964,8 @@ def test_windows_python3_gpu_mkldnn(lib_name) {
unstash lib_name
powershell 'ci/windows/test_py3_gpu.ps1'
} finally {
- utils.collect_test_results_windows('tests_forward.xml', 'tests_gpu_forward_windows_python3_gpu_mkldnn.xml')
- utils.collect_test_results_windows('tests_operator.xml', 'tests_gpu_operator_windows_python3_gpu_mkldnn.xml')
+ utils.collect_test_results_windows('tests_forward.xml', 'tests_gpu_forward_windows_python3_gpu_onednn.xml')
+ utils.collect_test_results_windows('tests_operator.xml', 'tests_gpu_operator_windows_python3_gpu_onednn.xml')
}
}
}
diff --git a/ci/jenkins/Jenkinsfile_centos_cpu b/ci/jenkins/Jenkinsfile_centos_cpu
index cddb3c9..f06b301 100644
--- a/ci/jenkins/Jenkinsfile_centos_cpu
+++ b/ci/jenkins/Jenkinsfile_centos_cpu
@@ -35,7 +35,7 @@ utils.main_wrapper(
core_logic: {
utils.parallel_stage('Build', [
custom_steps.compile_centos7_cpu('centos7_cpu'),
- custom_steps.compile_centos7_cpu_mkldnn(),
+ custom_steps.compile_centos7_cpu_onednn(),
custom_steps.compile_static_python_cpu(),
custom_steps.compile_static_cd_cpu('centos7_cpu_cd')
])
diff --git a/ci/jenkins/Jenkinsfile_clang b/ci/jenkins/Jenkinsfile_clang
index 953d4b4..4fecc6f 100644
--- a/ci/jenkins/Jenkinsfile_clang
+++ b/ci/jenkins/Jenkinsfile_clang
@@ -37,8 +37,8 @@ core_logic: {
custom_steps.compile_unix_clang_6_cpu(),
custom_steps.compile_unix_clang_10_cpu(),
custom_steps.compile_unix_clang_tidy_cpu(),
- custom_steps.compile_unix_clang_6_mkldnn_cpu(),
- custom_steps.compile_unix_clang_10_mkldnn_cpu()
+ custom_steps.compile_unix_clang_6_onednn_cpu(),
+ custom_steps.compile_unix_clang_10_onednn_cpu()
])
}
,
diff --git a/ci/jenkins/Jenkinsfile_unix_cpu b/ci/jenkins/Jenkinsfile_unix_cpu
index 2b2f5bc..7cc70d4 100644
--- a/ci/jenkins/Jenkinsfile_unix_cpu
+++ b/ci/jenkins/Jenkinsfile_unix_cpu
@@ -37,8 +37,8 @@ core_logic: {
custom_steps.compile_unix_cpu_openblas('cpu'),
custom_steps.compile_unix_openblas_debug_cpu('cpu_debug'),
custom_steps.compile_unix_mkl_cpu('cpu_mkl'),
- custom_steps.compile_unix_mkldnn_cpu('mkldnn_cpu'),
- custom_steps.compile_unix_mkldnn_mkl_cpu('mkldnn_mkl_cpu'),
+ custom_steps.compile_unix_onednn_cpu('onednn_cpu'),
+ custom_steps.compile_unix_onednn_mkl_cpu('onednn_mkl_cpu'),
custom_steps.compile_unix_int64_cpu('ubuntu_cpu'),
custom_steps.compile_unix_openblas_cpu_no_tvm_op('cpu_openblas_no_tvm_op'),
])
@@ -46,8 +46,8 @@ core_logic: {
utils.parallel_stage('Tests', [
custom_steps.test_unix_python3_cpu('cpu'),
custom_steps.test_unix_python3_mkl_cpu('cpu_mkl'),
- custom_steps.test_unix_python3_mkldnn_cpu('mkldnn_cpu'),
- custom_steps.test_unix_python3_mkldnn_mkl_cpu('mkldnn_mkl_cpu'),
+ custom_steps.test_unix_python3_onednn_cpu('onednn_cpu'),
+ custom_steps.test_unix_python3_onednn_mkl_cpu('onednn_mkl_cpu'),
/* disable onnx tests for now, until onnx work is forwarded-ported to master
custom_steps.test_unix_onnx_cpu('cpu'),
*/
diff --git a/ci/jenkins/Jenkinsfile_unix_gpu b/ci/jenkins/Jenkinsfile_unix_gpu
index 63bd3d2..ef385af 100644
--- a/ci/jenkins/Jenkinsfile_unix_gpu
+++ b/ci/jenkins/Jenkinsfile_unix_gpu
@@ -34,8 +34,8 @@ utils.assign_node_labels(utility: 'utility', linux_cpu: 'mxnetlinux-cpu', linux_
utils.main_wrapper(
core_logic: {
utils.parallel_stage('Build', [
- custom_steps.compile_unix_mkldnn_gpu('mkldnn_gpu'),
- custom_steps.compile_unix_mkldnn_nocudnn_gpu('mkldnn_gpu_nocudnn'),
+ custom_steps.compile_unix_onednn_gpu('onednn_gpu'),
+ custom_steps.compile_unix_onednn_nocudnn_gpu('onednn_gpu_nocudnn'),
custom_steps.compile_unix_full_gpu('gpu'),
custom_steps.compile_unix_full_gpu_debug('gpu_debug'),
custom_steps.compile_unix_tensorrt_gpu('tensorrt'),
@@ -44,8 +44,8 @@ core_logic: {
utils.parallel_stage('Tests', [
custom_steps.test_unix_python3_gpu('gpu'),
- custom_steps.test_unix_python3_mkldnn_gpu('mkldnn_gpu'),
- custom_steps.test_unix_python3_mkldnn_nocudnn_gpu('mkldnn_gpu_nocudnn'),
+ custom_steps.test_unix_python3_onednn_gpu('onednn_gpu'),
+ custom_steps.test_unix_python3_onednn_nocudnn_gpu('onednn_gpu_nocudnn'),
// TODO(szha): fix and reenable the hanging issue. tracked in #18098
// custom_steps.test_unix_distributed_kvstore_gpu('gpu'),
custom_steps.test_unix_byteps_gpu('gpu'),
diff --git a/ci/jenkins/Jenkinsfile_windows_cpu b/ci/jenkins/Jenkinsfile_windows_cpu
index 2592f65..63e1a91 100644
--- a/ci/jenkins/Jenkinsfile_windows_cpu
+++ b/ci/jenkins/Jenkinsfile_windows_cpu
@@ -35,8 +35,8 @@ utils.main_wrapper(
core_logic: {
utils.parallel_stage('Build', [
custom_steps.compile_windows_cpu('windows_package_cpu'),
- custom_steps.compile_windows_cpu_mkldnn('windows_package_cpu_mkldnn'),
- custom_steps.compile_windows_cpu_mkldnn_mkl('windows_package_cpu_mkldnn_mkl'),
+ custom_steps.compile_windows_cpu_onednn('windows_package_cpu_onednn'),
+ custom_steps.compile_windows_cpu_onednn_mkl('windows_package_cpu_onednn_mkl'),
custom_steps.compile_windows_cpu_mkl('windows_package_cpu_mkl')
])
diff --git a/ci/jenkins/Jenkinsfile_windows_gpu b/ci/jenkins/Jenkinsfile_windows_gpu
index 1c4a9c9..3e8e889 100644
--- a/ci/jenkins/Jenkinsfile_windows_gpu
+++ b/ci/jenkins/Jenkinsfile_windows_gpu
@@ -35,12 +35,12 @@ utils.main_wrapper(
core_logic: {
utils.parallel_stage('Build', [
custom_steps.compile_windows_gpu('windows_package_gpu'),
- custom_steps.compile_windows_gpu_mkldnn('windows_package_gpu_mkldnn')
+ custom_steps.compile_windows_gpu_onednn('windows_package_gpu_onednn')
])
utils.parallel_stage('Tests', [
custom_steps.test_windows_python3_gpu('windows_package_gpu'),
- custom_steps.test_windows_python3_gpu_mkldnn('windows_package_gpu_mkldnn')
+ custom_steps.test_windows_python3_gpu_onednn('windows_package_gpu_onednn')
])
}
,
diff --git a/tests/nightly/JenkinsfileForBinaries b/tests/nightly/JenkinsfileForBinaries
index 637180f..18dcb83 100755
--- a/tests/nightly/JenkinsfileForBinaries
+++ b/tests/nightly/JenkinsfileForBinaries
@@ -32,9 +32,9 @@ core_logic: {
stage('Build') {
parallel 'GPU: CUDA10.1+cuDNN7': {
node(NODE_LINUX_CPU) {
- ws('workspace/build-mkldnn-gpu') {
+ ws('workspace/build-onednn-gpu') {
utils.init_git()
- utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_mkldnn', false)
+ utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_onednn', false)
utils.pack_lib('gpu', mx_lib)
}
}