From 2092f6df05960c52d9df1994b0af812c2fc6544a Mon Sep 17 00:00:00 2001 From: Mohammad Miadh Angkad <176301910+mmangkad@users.noreply.github.com> Date: Thu, 10 Sep 2026 07:00:17 +0800 Subject: [PATCH] [CI] Install helion 1.4.0 for the KDA Helion kernel tests (#38688) Co-authored-by: Mohammad Angkad Co-authored-by: Alison Shao <54658187+alisonshao@users.noreply.github.com> --- scripts/ci/cuda/ci_install_dependency.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/ci/cuda/ci_install_dependency.sh b/scripts/ci/cuda/ci_install_dependency.sh index 067c64c0a..c1afa9ed5 100755 --- a/scripts/ci/cuda/ci_install_dependency.sh +++ b/scripts/ci/cuda/ci_install_dependency.sh @@ -761,6 +761,8 @@ install_extra_deps() { fi $PIP_CMD install ${MOONCAKE_PKG} ${EXTRA_NVIDIA_SPECS} py-spy scipy huggingface_hub[hf_xet] pytest $PIP_INSTALL_SUFFIX + $PIP_CMD install "helion==1.4.0" $PIP_INSTALL_SUFFIX + NIXL_INSTALLED=$(pip show nixl 2>/dev/null | grep "^Version:" | awk '{print $2}' || echo "") NIXL_BIN_INSTALLED=$(pip show "${NIXL_BIN_NAME}" 2>/dev/null | grep "^Version:" | awk '{print $2}' || echo "") if [ "$NIXL_INSTALLED" = "$NIXL_VERSION" ] && [ "$NIXL_BIN_INSTALLED" = "$NIXL_VERSION" ]; then