diff --git a/ci/build_windows.py b/ci/build_windows.py index 2035e4c83bf0..fe2a03b03c09 100755 --- a/ci/build_windows.py +++ b/ci/build_windows.py @@ -48,11 +48,11 @@ 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 @@ class BuildFlavour(Enum): '-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 @@ class BuildFlavour(Enum): '-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 @@ class BuildFlavour(Enum): '-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 319d8a72793c..9f0f928f60b3 100644 --- a/ci/dev_menu.py +++ b/ci/dev_menu.py @@ -141,14 +141,14 @@ def provision_virtualenv(venv_path=DEFAULT_PYENV): "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 078b4ffb7df6..e53a57bde3ec 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 ac30fffcbc2a..ee6a0a27e502 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_mkldnn_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_mkldnn_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_mkldnn_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_mkldnn_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_mkldnn_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_mkldnn_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_mkldnn_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 cddb3c977a5d..f06b30101941 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 953d4b43653d..4fecc6f69a52 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 2b2f5bc0271a..7cc70d40c22c 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 63bd3d29b641..ef385af2078d 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 2592f65c6b90..63e1a91e791b 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 1c4a9c9988b6..3e8e889aef44 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 637180ff8916..18dcb839748b 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) } }