Skip to content

[Fix]: mla_decode test aborts whole collection on aiter API drift - #840

Merged
coderfeli merged 1 commit into
ROCm:mainfrom
Phil-amd:fix/mla-decode-aiter-import
Jul 13, 2026
Merged

coderfeli merged 1 commit into
ROCm:mainfrom
Phil-amd:fix/mla-decode-aiter-import

Conversation

@Phil-amd

Copy link
Copy Markdown
Member

importorskip only covers a missing aiter module; the top-level symbol import still hard-fails when the attention API drifts (e.g. hk_mla_decode_fwd renamed), raising ImportError during collection and taking down the entire test session (exit 2). Wrap it in try/except and skip the module instead so the mismatch degrades to a clean per-module skip.

importorskip only covers a missing aiter module; the top-level symbol
import still hard-fails when the attention API drifts (e.g.
hk_mla_decode_fwd renamed), raising ImportError during collection and
taking down the entire test session (exit 2). Wrap it in try/except and
skip the module instead so the mismatch degrades to a clean per-module
skip.
@coderfeli
coderfeli merged commit 99284f7 into ROCm:main Jul 13, 2026
6 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants