File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change 83
83
VLLM_WORKER_MULTIPROC_METHOD : spawn
84
84
VLLM_USE_MODELSCOPE : True
85
85
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
86
89
if : ${{ inputs.type == 'full' }}
87
90
run : |
88
91
# We found that if running aclgraph tests in batch, it will cause AclmdlRICaptureBegin error. So we run
@@ -103,8 +106,7 @@ jobs:
103
106
# ------------------------------------ v1 spec decode test ------------------------------------ #
104
107
pytest -sv tests/e2e/singlecard/spec_decode_v1/test_v1_mtp_correctness.py
105
108
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
108
110
109
111
pytest -sv tests/e2e/singlecard/ops/
110
112
You can’t perform that action at this time.
0 commit comments