Skip to content

Commit 3ba0256

Browse files
committed
Test eagle OOM
Signed-off-by: Icey <[email protected]>
1 parent 2dde126 commit 3ba0256

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

.github/workflows/_e2e_test.yaml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,9 @@ jobs:
8383
VLLM_WORKER_MULTIPROC_METHOD: spawn
8484
VLLM_USE_MODELSCOPE: True
8585
PYTORCH_NPU_ALLOC_CONF: max_split_size_mb:256
86+
PYTORCH_NPU_ALLOC_CONF: expandable_segments:True
87+
NPU_MEMORY_FRACTION: 0.96
88+
OMP_NUM_THREADS: 1
8689
if: ${{ inputs.type == 'full' }}
8790
run: |
8891
# We found that if running aclgraph tests in batch, it will cause AclmdlRICaptureBegin error. So we run
@@ -103,8 +106,7 @@ jobs:
103106
# ------------------------------------ v1 spec decode test ------------------------------------ #
104107
pytest -sv tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py
105108
pytest -sv tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_torchair_correctness.py
106-
# Fix me: OOM error
107-
#pytest -sv tests/e2e/singlecard/spec_decode_v1/test_v1_spec_decode.py
109+
pytest -sv tests/e2e/singlecard/spec_decode_v1/test_v1_spec_decode.py
108110
109111
pytest -sv tests/e2e/singlecard/ops/
110112

0 commit comments

Comments
 (0)