[Build/CI] Build arm64 PR and postmerge image builds for Blackwell SM10x and SM110 (#48041)
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
Tyler Michael Smith committed
Jul 10, 2026 at 10:12 UTC
978a6dfa3f79470e4459a454816202e78c1d2108
1 file changed
+3
-2
.buildkite/image_build/image_build_arm64.sh
+3
-2
@@ -19,13 +19,14 @@ if docker manifest inspect "$IMAGE" >/dev/null 2>&1; then
19
echo "Image found"
20
else
21
echo "Image not found, proceeding with build..."
22
- # build for arm64 GPU targets: Grace/GH200 (sm_90) and DGX Spark/GB10
22
+ # build for arm64 GPU targets: Grace/GH200 (sm_90),
23
+ # Blackwell/Thor (sm_100/sm_103/sm_110), and DGX Spark/GB10
24
# (sm_121, family-covered by 12.0 under CUDA 13)
25
docker build --file docker/Dockerfile \
26
--platform linux/arm64 \
27
--build-arg max_jobs=16 \
28
--build-arg nvcc_threads=4 \
28
- --build-arg torch_cuda_arch_list="9.0 12.0" \
29
+ --build-arg torch_cuda_arch_list="9.0 10.0 11.0 12.0" \
30
--build-arg USE_SCCACHE=1 \
31
--build-arg buildkite_commit="$BUILDKITE_COMMIT" \
32
--tag "$IMAGE" \