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

Add publish test of PyPi cu100mkl #14637

Merged
merged 11 commits into from
Apr 18, 2019
Merged
Show file tree
Hide file tree
Changes from 9 commits
Commits
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
2 changes: 1 addition & 1 deletion ci/docker/Dockerfile.publish.ubuntu1404_gpu
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
#
# Dockerfile to run MXNet on Ubuntu 14.04 for GPU

FROM ubuntu:14.04
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu14.04
stu1130 marked this conversation as resolved.
Show resolved Hide resolved

WORKDIR /work/deps

Expand Down
10 changes: 9 additions & 1 deletion ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -1327,7 +1327,7 @@ deploy_jl_docs() {
# ...
}

build_scala_static_mkl() {
build_static_scala_mkl() {
set -ex
pushd .
scala_prepare
Expand All @@ -1345,6 +1345,14 @@ build_static_python_mkl() {
popd
}

build_static_python_cu100mkl() {
set -ex
pushd .
export mxnet_variant=cu100mkl
./ci/publish/python/build.sh
popd
}

publish_scala_build() {
set -ex
pushd .
Expand Down
15 changes: 14 additions & 1 deletion ci/jenkins/Jenkins_steps.groovy
Original file line number Diff line number Diff line change
Expand Up @@ -521,7 +521,7 @@ def test_static_scala_cpu() {
ws('workspace/ut-publish-scala-cpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run("publish.ubuntu1404_cpu", 'build_scala_static_mkl', false)
utils.docker_run("publish.ubuntu1404_cpu", 'build_static_scala_mkl', false)
}
}
}
Expand All @@ -541,6 +541,19 @@ def test_static_python_cpu() {
}]
}

def test_static_python_gpu() {
return ['Static build GPU 14.04 Python' : {
node(NODE_LINUX_GPU) {
ws('workspace/ut-publish-python-gpu') {
timeout(time: max_time, unit: 'MINUTES') {
utils.init_git()
utils.docker_run("publish.ubuntu1404_gpu", 'build_static_python_cu100mkl', true)
}
}
}
}]
}

def test_unix_python2_cpu() {
return ['Python2: CPU': {
node(NODE_LINUX_CPU) {
Expand Down
3 changes: 2 additions & 1 deletion ci/jenkins/Jenkinsfile_unix_gpu
Original file line number Diff line number Diff line change
Expand Up @@ -59,7 +59,8 @@ core_logic: {
custom_steps.test_unix_python3_integration_gpu(),
custom_steps.test_unix_cpp_package_gpu(),
custom_steps.test_unix_scala_gpu(),
custom_steps.test_unix_distributed_kvstore_gpu()
custom_steps.test_unix_distributed_kvstore_gpu(),
custom_steps.test_static_python_gpu()
stu1130 marked this conversation as resolved.
Show resolved Hide resolved

// Disabled due to: /~https://github.com/apache/incubator-mxnet/issues/11407
//custom_steps.test_unix_caffe_gpu()
Expand Down