Skip to content

Commit 2e5bce1

Browse files
authored
Merge branch 'master' into tf-inference-gpu-2.18-patch
2 parents ea170bf + cdf0eb3 commit 2e5bce1

File tree

3 files changed

+18
-4
lines changed

3 files changed

+18
-4
lines changed

miscellaneous_scripts/dockerfile_patch_script.sh

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,20 @@ if [[ $LATEST_RELEASED_IMAGE_URI =~ ^763104351884\.dkr\.ecr\.us-west-2\.amazonaw
6161
chmod +x /usr/local/bin/start_cuda_compat.sh
6262
fi
6363

64+
# For PT ARM64 GPU images, patch nvjpeg
65+
if [[ $LATEST_RELEASED_IMAGE_URI =~ ^763104351884\.dkr\.ecr\.us-west-2\.amazonaws\.com/pytorch-(inference|training)-arm64:2\.[5-7]\.[0-9]+-gpu(.+) ]]; then
66+
mkdir -p /tmp/nvjpeg
67+
cd /tmp/nvjpeg
68+
wget https://developer.download.nvidia.com/compute/cuda/redist/libnvjpeg/linux-aarch64/libnvjpeg-linux-aarch64-12.4.0.76-archive.tar.xz
69+
tar -xvf libnvjpeg-linux-aarch64-12.4.0.76-archive.tar.xz
70+
rm -rf /usr/local/cuda/targets/sbsa-linux/lib/libnvjpeg*
71+
rm -rf /usr/local/cuda/targets/sbsa-linux/include/nvjpeg.h
72+
cp libnvjpeg-linux-aarch64-12.4.0.76-archive/lib/libnvjpeg* /usr/local/cuda/targets/sbsa-linux/lib/
73+
cp libnvjpeg-linux-aarch64-12.4.0.76-archive/include/* /usr/local/cuda/targets/sbsa-linux/include/
74+
cd /
75+
rm -rf /tmp/nvjpeg
76+
fi
77+
6478
# For all GPU images, remove cuobjdump and nvdisasm
6579
if [[ $LATEST_RELEASED_IMAGE_URI =~ ^763104351884\.dkr\.ecr\.us-west-2\.amazonaws\.com/(pytorch|tensorflow)(.+)gpu(.+) ]]; then
6680
rm -rf /usr/local/cuda/bin/cuobjdump*

pytorch/training/buildspec-arm64-2-7-ec2.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ framework: &FRAMEWORK pytorch
55
version: &VERSION 2.7.0
66
short_version: &SHORT_VERSION "2.7"
77
arch_type: arm64
8-
autopatch_build: "False"
8+
autopatch_build: "True"
99

1010
repository_info:
1111
training_repository: &TRAINING_REPOSITORY

release_images_inference.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -185,13 +185,13 @@ release_images:
185185
force_release: False
186186
15:
187187
framework: "djl"
188-
version: "0.33.0"
188+
version: "0.0.0.dev1"
189189
arch_type: "x86"
190190
inference:
191191
device_types: [ "gpu" ]
192192
python_versions: [ "py312" ]
193193
os_version: "ubuntu24.04"
194-
tensorrtllm_version: "0.21.0"
194+
lmi_version: "0.0.0"
195195
cuda_version: "cu128"
196196
example: False
197197
disable_sm_tag: True
@@ -222,4 +222,4 @@ release_images:
222222
cuda_version: "cu128"
223223
example: False
224224
disable_sm_tag: False
225-
force_release: False
225+
force_release: False

0 commit comments

Comments
 (0)