[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:
co-authored by
Mohammad Angkad
Alison Shao
parent
51c8581a26
commit
2092f6df05
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user