Lines Matching +full:build +full:- +full:docker +full:- +full:xpu
3 set -ex
8 if [ -z "${image}" ]; then
14 eval export $2=$(echo "${image}" | perl -n -e"/$1(\d+(\.\d+)?(\.\d+)?)/ && print \$1")
22 # parts $image into array, splitting on '-'
24 IFS="-"
25 declare -a parts=($image)
30 name=$(echo "${part}" | perl -n -e"/([a-zA-Z]+)\d+(\.\d+)?(\.\d+)?/ && print \$1")
36 # skip non-conforming fields such as "pytorch", "linux" or "bionic" without version string
37 if [ -n "${name}" ]; then
43 # Use the same pre-built XLA test image from PyTorch/XLA
45 echo "Using pre-built XLA test image..."
49 if [[ "$image" == *-focal* ]]; then
51 elif [[ "$image" == *-jammy* ]]; then
59 if [ -n "${UBUNTU_VERSION}" ]; then
61 elif [ -n "${CENTOS_VERSION}" ]; then
69 # When using ubuntu - 22.04, start from Ubuntu docker image, instead of nvidia/cuda docker image.
71 DOCKERFILE="${OS}-cuda/Dockerfile"
73 DOCKERFILE="${OS}-rocm/Dockerfile"
74 elif [[ "$image" == *xpu* ]]; then
75 DOCKERFILE="${OS}-xpu/Dockerfile"
78 DOCKERFILE="linter-cuda/Dockerfile"
94 pytorch-linux-focal-cuda12.4-cudnn9-py3-gcc9)
108 pytorch-linux-focal-cuda12.1-cudnn9-py3-gcc9)
122 pytorch-linux-focal-cuda12.4-cudnn9-py3-gcc9-inductor-benchmarks)
137 pytorch-linux-focal-cuda12.1-cudnn9-py3-gcc9-inductor-benchmarks)
152 pytorch-linux-focal-cuda12.1-cudnn9-py3.12-gcc9-inductor-benchmarks)
167 pytorch-linux-focal-cuda12.4-cudnn9-py3.12-gcc9-inductor-benchmarks)
182 pytorch-linux-focal-cuda11.8-cudnn9-py3-gcc9)
196 pytorch-linux-focal-cuda12.4-cudnn9-py3-gcc9)
210 pytorch-linux-focal-cuda12.1-cudnn9-py3-gcc9)
224 pytorch-linux-focal-cuda12.4-cudnn9-py3-gcc9)
238 pytorch-linux-focal-py3-clang10-onnx)
247 pytorch-linux-focal-py3-clang9-android-ndk-r21e)
257 pytorch-linux-focal-py3.9-clang10)
268 pytorch-linux-focal-py3.11-clang10)
279 pytorch-linux-focal-py3.9-gcc9)
288 pytorch-linux-focal-rocm-n-1-py3)
299 pytorch-linux-focal-rocm-n-py3)
310 pytorch-linux-jammy-xpu-2024.0-py3)
321 pytorch-linux-jammy-py3.9-gcc11-inductor-benchmarks)
333 pytorch-linux-jammy-cuda11.8-cudnn9-py3.9-clang12)
343 pytorch-linux-jammy-py3-clang12-asan)
352 pytorch-linux-jammy-py3-clang15-asan)
358 pytorch-linux-jammy-py3.9-gcc11)
370 pytorch-linux-jammy-py3-clang12-executorch)
376 pytorch-linux-jammy-py3.12-halide)
383 pytorch-linux-focal-linter)
390 pytorch-linux-jammy-cuda11.8-cudnn9-py3.9-linter)
395 pytorch-linux-jammy-aarch64-py3.10-gcc11)
406 # snadampal: skipping llvm src build install because the current version
410 pytorch-linux-jammy-aarch64-py3.10-gcc11-inductor-benchmarks)
421 # snadampal: skipping llvm src build install because the current version
427 # Catch-all for builds that are not hardcoded.
431 echo "image '$image' did not match an existing build configuration"
443 # To ensure that any ROCm config will build using conda cmake
468 tmp_tag=$(basename "$(mktemp -u)" | tr '[:upper:]' '[:lower:]')
472 IMAGE_NAME="nvidia/cuda:${CUDA_VERSION}-cudnn${CUDNN_VERSION}-devel-ubuntu${UBUNTU_VERSION}"
474 IMAGE_NAME="nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}"
478 # Build image
479 docker build \
480 --no-cache \
481 --progress=plain \
482 --build-arg "BUILD_ENVIRONMENT=${image}" \
483 --build-arg "PROTOBUF=${PROTOBUF:-}" \
484 --build-arg "LLVMDEV=${LLVMDEV:-}" \
485 --build-arg "DB=${DB:-}" \
486 --build-arg "VISION=${VISION:-}" \
487 --build-arg "UBUNTU_VERSION=${UBUNTU_VERSION}" \
488 --build-arg "CENTOS_VERSION=${CENTOS_VERSION}" \
489 --build-arg "DEVTOOLSET_VERSION=${DEVTOOLSET_VERSION}" \
490 --build-arg "GLIBC_VERSION=${GLIBC_VERSION}" \
491 --build-arg "CLANG_VERSION=${CLANG_VERSION}" \
492 --build-arg "ANACONDA_PYTHON_VERSION=${ANACONDA_PYTHON_VERSION}" \
493 --build-arg "GCC_VERSION=${GCC_VERSION}" \
494 --build-arg "CUDA_VERSION=${CUDA_VERSION}" \
495 --build-arg "CUDNN_VERSION=${CUDNN_VERSION}" \
496 --build-arg "TENSORRT_VERSION=${TENSORRT_VERSION}" \
497 --build-arg "ANDROID=${ANDROID}" \
498 --build-arg "ANDROID_NDK=${ANDROID_NDK_VERSION}" \
499 --build-arg "GRADLE_VERSION=${GRADLE_VERSION}" \
500 --build-arg "VULKAN_SDK_VERSION=${VULKAN_SDK_VERSION}" \
501 --build-arg "SWIFTSHADER=${SWIFTSHADER}" \
502 --build-arg "CMAKE_VERSION=${CMAKE_VERSION:-}" \
503 --build-arg "NINJA_VERSION=${NINJA_VERSION:-}" \
504 --build-arg "KATEX=${KATEX:-}" \
505 --build-arg "ROCM_VERSION=${ROCM_VERSION:-}" \
506 --build-arg "PYTORCH_ROCM_ARCH=${PYTORCH_ROCM_ARCH:-gfx906;gfx90a}" \
507 --build-arg "IMAGE_NAME=${IMAGE_NAME}" \
508 --build-arg "UCX_COMMIT=${UCX_COMMIT}" \
509 --build-arg "UCC_COMMIT=${UCC_COMMIT}" \
510 --build-arg "CONDA_CMAKE=${CONDA_CMAKE}" \
511 --build-arg "TRITON=${TRITON}" \
512 --build-arg "ONNX=${ONNX}" \
513 --build-arg "DOCS=${DOCS}" \
514 --build-arg "INDUCTOR_BENCHMARKS=${INDUCTOR_BENCHMARKS}" \
515 --build-arg "EXECUTORCH=${EXECUTORCH}" \
516 --build-arg "HALIDE=${HALIDE}" \
517 --build-arg "XPU_VERSION=${XPU_VERSION}" \
518 --build-arg "ACL=${ACL:-}" \
519 --build-arg "SKIP_SCCACHE_INSTALL=${SKIP_SCCACHE_INSTALL:-}" \
520 --build-arg "SKIP_LLVM_SRC_BUILD_INSTALL=${SKIP_LLVM_SRC_BUILD_INSTALL:-}" \
521 -f $(dirname ${DOCKERFILE})/Dockerfile \
522 -t "$tmp_tag" \
526 # NVIDIA dockers for RC releases use tag names like `11.0-cudnn9-devel-ubuntu18.04-rc`,
527 # for this case we will set UBUNTU_VERSION to `18.04-rc` so that the Dockerfile could
529 # "$UBUNTU_VERSION" == "18.04-rc"
532 UBUNTU_VERSION=$(echo ${UBUNTU_VERSION} | sed 's/-rc$//')
535 docker run --rm "$tmp_tag" $*
540 if !(drun lsb_release -a 2>&1 | grep -qF Ubuntu); then
542 drun lsb_release -a
545 if !(drun lsb_release -a 2>&1 | grep -qF "$UBUNTU_VERSION"); then
547 drun lsb_release -a
552 if [ -n "$ANACONDA_PYTHON_VERSION" ]; then
553 if !(drun python --version 2>&1 | grep -qF "Python $ANACONDA_PYTHON_VERSION"); then
555 drun python --version
560 if [ -n "$GCC_VERSION" ]; then
561 if !(drun gcc --version 2>&1 | grep -q " $GCC_VERSION\\W"); then
563 drun gcc --version
568 if [ -n "$CLANG_VERSION" ]; then
569 if !(drun clang --version 2>&1 | grep -qF "clang version $CLANG_VERSION"); then
571 drun clang --version
576 if [ -n "$KATEX" ]; then
577 if !(drun katex --version); then
579 drun katex --version