Skip to content

Commit bf4a60e

Browse files
authored
Revise backend test for kunlunxin (flagos-ai#2299)
1 parent 337137d commit bf4a60e

2 files changed

Lines changed: 2 additions & 12 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -83,19 +83,11 @@ iluvatar = [
8383
]
8484

8585
kunlunxin = [
86-
"flag_gems[official_torch_251]",
87-
"benchflow==1.0.0",
88-
"torch_klx==0.1.0",
89-
"torch_xray==0.2.1",
90-
"triton==3.0.0+0762702f",
91-
"xmlir==1.0.0.1",
92-
# "hydra==0.1.0",
93-
# "transformer_engine==1.13.0+7fc5d8b",
94-
# Other dependencies
9586
"regex==2026.4.4",
9687
"hyperparameter==0.5.6",
9788
"colorama==0.4.6",
9889
"psutil==6.1.0",
90+
"setuptools==81.0.0",
9991
]
10092

10193
mthreads = [

tools/run_backend_tests_kunlunxin.sh

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,7 @@ source .venv/bin/activate
2020
uv pip install setuptools==82.0.1 scikit-build-core==0.12.2 pybind11==3.0.3 cmake==3.31.10 ninja==1.13.0
2121

2222
uv pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index https://download.pytorch.org/whl/cu118
23-
uv pip install benchflow==1.0.0 torch_klx==0.1.0 torch_xray==0.2.1 triton==3.0.0+0762702f xmlir==1.0.0.1 \
24-
--index https://resource.flagos.net/repository/flagos-pypi-kunlunxin/simple
25-
23+
$HOME/kunlunxin/install-wheels.sh
2624
uv pip install -e .[kununxin,test]
2725

2826
TEST_FILES=(

0 commit comments

Comments
 (0)