[CI] Install helion 1.4.0 for the KDA Helion kernel tests (#38688)

Co-authored-by: Mohammad Angkad <mohammad.angkad@radixark.ai>
Co-authored-by: Alison Shao <54658187+alisonshao@users.noreply.github.com>
This commit is contained in:
Mohammad Miadh Angkad
2026-09-09 16:00:17 -07:00
committed by GitHub
co-authored by Mohammad Angkad Alison Shao
parent 51c8581a26
commit 2092f6df05
+2
View File
@@ -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