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

Commit

Permalink
Remove edge build, update cuda version to build and test with.
Browse files Browse the repository at this point in the history
  • Loading branch information
@josephevans
josephevans committed Sep 1, 2023
1 parent 2daabcd commit eb6159f
Showing 1 changed file with 15 additions and 15 deletions.
30 changes: 15 additions & 15 deletions ci/jenkins/Jenkins_steps.groovy
View file
Original file line number Diff line number Diff line change
Expand Up @@ -151,7 +151,7 @@ def compile_unix_int64_gpu(lib_name) {
ws('workspace/build-gpu-int64') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_large_tensor', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu_large_tensor', false)
utils.pack_lib(lib_name, mx_cmake_lib)
}
}
Expand Down Expand Up @@ -207,7 +207,7 @@ def compile_unix_onednn_gpu(lib_name) {
ws('workspace/build-onednn-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_onednn', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu_onednn', false)
utils.pack_lib(lib_name, mx_onednn_lib)
}
}
Expand All @@ -221,7 +221,7 @@ def compile_unix_onednn_nocudnn_gpu(lib_name) {
ws('workspace/build-onednn-gpu-nocudnn') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_onednn_nocudnn', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu_onednn_nocudnn', false)
utils.pack_lib(lib_name, mx_onednn_lib)
}
}
Expand All @@ -235,7 +235,7 @@ def compile_unix_full_gpu(lib_name) {
ws('workspace/build-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu', false)
utils.pack_lib(lib_name, mx_lib_cpp_examples)
}
}
Expand All @@ -249,7 +249,7 @@ def compile_unix_full_gpu_debug(lib_name) {
ws('workspace/build-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_debug', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu_debug', false)
utils.pack_lib(lib_name, mx_lib_cpp_examples)
}
}
Expand Down Expand Up @@ -508,7 +508,7 @@ def compile_unix_clang10_cuda_werror(lib_name) {
ws('workspace/build-cpu-clang10') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run('ubuntu_gpu_cu111', 'build_ubuntu_gpu_clang10_werror', false)
utils.docker_run('ubuntu_gpu_cu118', 'build_ubuntu_gpu_clang10_werror', false)
utils.pack_lib(lib_name, mx_lib)
}
}
Expand Down Expand Up @@ -706,7 +706,7 @@ def test_unix_python3_gpu(lib_name) {
ws('workspace/ut-python3-gpu') {
try {
utils.unpack_and_init(lib_name, mx_lib_cython)
python3_gpu_ut_cython('ubuntu_gpu_cu111')
python3_gpu_ut_cython('ubuntu_gpu_cu118')
utils.publish_test_coverage()
} finally {
utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_gpu.xml')
Expand All @@ -722,7 +722,7 @@ def test_unix_python3_ampere_gpu(lib_name) {
ws('workspace/ut-python3-gpu') {
try {
utils.unpack_and_init(lib_name, mx_lib_cython)
python3_gpu_ut_cython('ubuntu_gpu_cu111')
python3_gpu_ut_cython('ubuntu_gpu_cu118')
utils.publish_test_coverage()
} finally {
utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_ampere_gpu.xml')
Expand Down Expand Up @@ -820,7 +820,7 @@ def test_unix_python3_onednn_gpu(lib_name) {
ws('workspace/ut-python3-onednn-gpu') {
try {
utils.unpack_and_init(lib_name, mx_onednn_lib)
python3_gpu_ut('ubuntu_gpu_cu111')
python3_gpu_ut('ubuntu_gpu_cu118')
utils.publish_test_coverage()
} finally {
utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_onednn_gpu.xml')
Expand All @@ -836,7 +836,7 @@ def test_unix_python3_onednn_nocudnn_gpu(lib_name) {
ws('workspace/ut-python3-onednn-gpu-nocudnn') {
try {
utils.unpack_and_init(lib_name, mx_onednn_lib)
python3_gpu_ut_nocudnn('ubuntu_gpu_cu111')
python3_gpu_ut_nocudnn('ubuntu_gpu_cu118')
utils.publish_test_coverage()
} finally {
utils.collect_test_results_unix('tests_gpu.xml', 'tests_python3_onednn_gpu_nocudnn.xml')
Expand Down Expand Up @@ -880,7 +880,7 @@ def test_unix_byteps_gpu(lib_name) {
ws('workspace/it-byteps') {
timeout(time: max_time, unit: 'MINUTES') {
utils.unpack_and_init(lib_name, mx_lib)
utils.docker_run('ubuntu_gpu_cu111', 'integrationtest_ubuntu_gpu_byteps', true, '32768m')
utils.docker_run('ubuntu_gpu_cu118', 'integrationtest_ubuntu_gpu_byteps', true, '32768m')
utils.publish_test_coverage()
}
}
Expand All @@ -894,7 +894,7 @@ def test_unix_distributed_kvstore_gpu(lib_name) {
ws('workspace/it-dist-kvstore') {
timeout(time: max_time, unit: 'MINUTES') {
utils.unpack_and_init(lib_name, mx_lib)
utils.docker_run('ubuntu_gpu_cu111', 'integrationtest_ubuntu_gpu_dist_kvstore', true)
utils.docker_run('ubuntu_gpu_cu118', 'integrationtest_ubuntu_gpu_dist_kvstore', true)
utils.publish_test_coverage()
}
}
Expand All @@ -908,7 +908,7 @@ def test_unix_cpp_package_gpu(lib_name) {
ws('workspace/it-cpp-package-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.unpack_and_init(lib_name, mx_lib_cpp_examples)
utils.docker_run('ubuntu_gpu_cu111', 'integrationtest_ubuntu_cpp_package_gpu', true)
utils.docker_run('ubuntu_gpu_cu118', 'integrationtest_ubuntu_cpp_package_gpu', true)
utils.publish_test_coverage()
}
}
Expand All @@ -922,7 +922,7 @@ def test_unix_python3_data_interchange_gpu(lib_name) {
ws('workspace/it-data-interchange') {
timeout(time: max_time, unit: 'MINUTES') {
utils.unpack_and_init(lib_name, mx_lib)
utils.docker_run('ubuntu_gpu_cu111', 'test_python3_data_interchange_gpu', true)
utils.docker_run('ubuntu_gpu_cu118', 'test_python3_data_interchange_gpu', true)
utils.publish_test_coverage()
}
}
Expand Down Expand Up @@ -1122,7 +1122,7 @@ def docs_python(lib_name) {
ws('workspace/docs') {
timeout(time: max_time, unit: 'MINUTES') {
utils.unpack_and_init(lib_name, mx_lib_cython)
utils.docker_run('ubuntu_gpu_cu111', 'build_python_docs', true)
utils.docker_run('ubuntu_gpu_cu118', 'build_python_docs', true)
if (should_pack_website()) {
utils.pack_lib('python-artifacts', 'docs/_build/python-artifacts.tgz', false)
}
Expand Down

0 comments on commit eb6159f

Please sign in to comment.