[Test] Add flashmla attention backend test (#5587)

2025-05-06 01:32:02 +08:00
parent 56f6589ecb
commit b8559764f6
3 changed files with 68 additions and 0 deletions
--- a/scripts/ci_install_dependency.sh
+++ b/scripts/ci_install_dependency.sh
@@ -31,3 +31,6 @@ pip install cuda-python nvidia-cuda-nvrtc-cu12
 # For lmms_evals evaluating MMMU
 git clone --branch v0.3.3 --depth 1 https://github.com/EvolvingLMMs-Lab/lmms-eval.git
 pip install -e lmms-eval/
+
+# Install FlashMLA for attention backend tests
+pip install git+https://github.com/deepseek-ai/FlashMLA.git