| #!/usr/bin/env bash |
| |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| # Docker bash script used to execute a command within a container. |
| # |
| if [ "$#" -lt 1 ]; then |
| echo "Usage: tests/script/docker_bash.sh <CONTAINER_NAME> [COMMAND]" |
| exit -1 |
| fi |
| |
| DOCKER_IMAGE_NAME=("$1") |
| |
| if [ "$#" -eq 1 ]; then |
| COMMAND="bash" |
| if [[ $(uname) == "Darwin" ]]; then |
| # Docker's host networking driver isn't supported on macOS. |
| # Use default bridge network and expose port for jupyter notebook. |
| CI_DOCKER_EXTRA_PARAMS=("-it -p 8888:8888") |
| else |
| CI_DOCKER_EXTRA_PARAMS=("-it --net=host") |
| fi |
| else |
| shift 1 |
| COMMAND=("$@") |
| fi |
| |
| SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" |
| WORKSPACE="$(pwd)" |
| |
| # Use nvidia-docker if the container is GPU. |
| if [[ ! -z $CUDA_VISIBLE_DEVICES ]]; then |
| CUDA_ENV="-e CUDA_VISIBLE_DEVICES=${CUDA_VISIBLE_DEVICES}" |
| else |
| CUDA_ENV="" |
| fi |
| |
| if [[ "${DOCKER_IMAGE_NAME}" == *"gpu"* ]]; then |
| if ! type "nvidia-docker" 1> /dev/null 2> /dev/null |
| then |
| DOCKER_BINARY="docker" |
| CUDA_ENV=" --gpus all "${CUDA_ENV} |
| else |
| DOCKER_BINARY="nvidia-docker" |
| fi |
| else |
| DOCKER_BINARY="docker" |
| fi |
| |
| # Print arguments. |
| echo "WORKSPACE: ${WORKSPACE}" |
| echo "DOCKER CONTAINER NAME: ${DOCKER_IMAGE_NAME}" |
| echo "" |
| |
| # FIXME(zhanghao): re-enable the tsim test after ISA is updated |
| if [[ ${COMMAND[@]} == "./tests/scripts/task_python_vta_tsim.sh" ]]; then |
| echo "Skip '${COMMAND[@]}'" |
| exit |
| fi |
| |
| echo "Running '${COMMAND[@]}' inside ${DOCKER_IMAGE_NAME}..." |
| |
| # By default we cleanup - remove the container once it finish running (--rm) |
| # and share the PID namespace (--pid=host) so the process inside does not have |
| # pid 1 and SIGKILL is propagated to the process inside (jenkins can kill it). |
| ${DOCKER_BINARY} run --rm --pid=host\ |
| -v ${WORKSPACE}:/workspace \ |
| -v ${SCRIPT_DIR}:/docker \ |
| -w /workspace \ |
| -e "CI_BUILD_HOME=/workspace" \ |
| -e "CI_BUILD_USER=$(id -u -n)" \ |
| -e "CI_BUILD_UID=$(id -u)" \ |
| -e "CI_BUILD_GROUP=$(id -g -n)" \ |
| -e "CI_BUILD_GID=$(id -g)" \ |
| -e "PYTHONPATH=python:topi/python"\ |
| ${CUDA_ENV}\ |
| ${CI_DOCKER_EXTRA_PARAMS[@]} \ |
| ${DOCKER_IMAGE_NAME}\ |
| bash --login tests/scripts/with_the_same_user \ |
| ${COMMAND[@]} |