Set up Vulkan SDK for VGF wheel builds

Signed-off-by: Rob Elliott <Robert.Elliott@arm.com>
Change-Id: I5680de25e7a004ac6883bd4f855baee1bf6b4874

Author: robell

.ci/scripts/setup-vulkan-macos-deps.sh

@@ -0,0 +1,63 @@
+#!/bin/bash
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
+# This source code is licensed under the BSD-style license found in the
+# LICENSE file in the root directory of this source tree.
+
+set -ex
+
+install_vulkan_sdk() {
+  VULKAN_SDK_VERSION=$1
+  _vulkan_sdk_url="https://sdk.lunarg.com/sdk/download/${VULKAN_SDK_VERSION}/mac/vulkansdk-macos-${VULKAN_SDK_VERSION}.zip"
+  _vulkan_sdk_dir=/tmp/vulkansdk
+  _vulkan_sdk_extract_dir=/tmp/vulkansdk-installer
+
+  mkdir -p "${_vulkan_sdk_dir}"
+  rm -rf "${_vulkan_sdk_extract_dir}"
+  mkdir -p "${_vulkan_sdk_extract_dir}"
+
+  _tmp_archive="/tmp/vulkansdk-macos.zip"
+
+  curl --silent --show-error --location --fail --retry 3 \
+    --output "${_tmp_archive}" "${_vulkan_sdk_url}"
+
+  unzip -q -o "${_tmp_archive}" -d "${_vulkan_sdk_extract_dir}"
+
+  _vulkan_sdk_app_path="$(
+    find "${_vulkan_sdk_extract_dir}" -maxdepth 3 -type d \
+      \( -name "vulkansdk-macOS-${VULKAN_SDK_VERSION}.app" \
+      -o -name "vulkansdk-macos-${VULKAN_SDK_VERSION}.app" \) \
+      -print -quit
+  )"
+  if [[ -z "${_vulkan_sdk_app_path}" ]]; then
+    echo "Failed to find Vulkan SDK installer app"
+    exit 1
+  fi
+  _vulkan_sdk_installer="${_vulkan_sdk_app_path}/Contents/MacOS/$(basename "${_vulkan_sdk_app_path}" .app)"
+  if [[ ! -x "${_vulkan_sdk_installer}" ]]; then
+    echo "Failed to find Vulkan SDK installer binary at ${_vulkan_sdk_installer}"
+    exit 1
+  fi
+
+  "${_vulkan_sdk_installer}" \
+    --root "${_vulkan_sdk_dir}/${VULKAN_SDK_VERSION}" \
+    --accept-licenses \
+    --default-answer \
+    --confirm-command install
+
+  export PATH="${PATH}:${_vulkan_sdk_dir}/${VULKAN_SDK_VERSION}/macOS/bin/"
+  export VULKAN_SDK="${_vulkan_sdk_dir}/${VULKAN_SDK_VERSION}/macOS"
+  export DYLD_LIBRARY_PATH="${_vulkan_sdk_dir}/${VULKAN_SDK_VERSION}/macOS/lib${DYLD_LIBRARY_PATH:+:${DYLD_LIBRARY_PATH}}"
+
+  _moltenvk_icd="${VULKAN_SDK}/share/vulkan/icd.d/MoltenVK_icd.json"
+  if [[ -f "${_moltenvk_icd}" ]]; then
+    export VK_DRIVER_FILES="${_moltenvk_icd}"
+  fi
+}
+
+VULKAN_SDK_VERSION="1.4.321.0"
+
+install_vulkan_sdk "${VULKAN_SDK_VERSION}"

.ci/scripts/wheel/pre_build_script.sh

@@ -2,6 +2,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
 #
+# Copyright 2026 Arm Limited and/or its affiliates.
+#
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
@@ -57,13 +59,37 @@ fi
 
 "${GITHUB_WORKSPACE}/${REPOSITORY}/install_requirements.sh" --example
 
+setup_vulkan_for_vgf() {
+  if [[ "$UNAME_S" == "Linux" && "$(uname -m)" == "x86_64" ]]; then
+    source "${GITHUB_WORKSPACE}/${REPOSITORY}/.ci/scripts/setup-vulkan-linux-deps.sh"
+  elif [[ "$UNAME_S" == "Darwin" ]]; then
+    source "${GITHUB_WORKSPACE}/${REPOSITORY}/.ci/scripts/setup-vulkan-macos-deps.sh"
+  fi
+
+  if ! command -v glslc >/dev/null 2>&1; then
+    return 1
+  fi
+
+  for _var in PATH VULKAN_SDK LD_LIBRARY_PATH DYLD_LIBRARY_PATH VK_ICD_FILENAMES VK_DRIVER_FILES ETVK_USING_SWIFTSHADER; do
+    if [[ -n "${!_var:-}" ]]; then
+      echo "${_var}=${!_var}" >> "${GITHUB_ENV}"
+    fi
+  done
+
+  return 0
+}
+
 # Enable VGF in pybind wheel builds when the platform-specific build input is
 # available from pip.
 if [[ "$UNAME_S" == "Linux" || "$UNAME_S" == "Darwin" ]]; then
   if python3 -m pip install -r \
     "${GITHUB_WORKSPACE}/${REPOSITORY}/backends/arm/requirements-arm-vgf-runtime.txt"; then
-    export EXECUTORCH_PYBIND_ENABLE_VGF=ON
-    echo "EXECUTORCH_PYBIND_ENABLE_VGF=ON" >> "${GITHUB_ENV}"
+    if setup_vulkan_for_vgf; then
+      export EXECUTORCH_PYBIND_ENABLE_VGF=ON
+      echo "EXECUTORCH_PYBIND_ENABLE_VGF=ON" >> "${GITHUB_ENV}"
+    else
+      echo "glslc unavailable after VGF build dependency install; building without VGF"
+    fi
   else
     echo "VGF build dependency unavailable on this platform; building without VGF"
   fi

tools/cmake/preset/pybind.cmake

@@ -1,5 +1,6 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # All rights reserved.
+# Copyright 2026 Arm Limited and/or its affiliates.
 #
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.