[CI] Re-enable sleep mode test and skip failure breaking CI (#990)

### What this PR does / why we need it?

- Re-enable sleep mode test
- Fix nightly performance benchmark workflow
- Fix model-runner-v1 bug for upstream
[change](https://github.com/vllm-project/vllm/pull/18654)
---------

Signed-off-by: wangli <wangli858794774@gmail.com>
This commit is contained in:
Li Wang
2025-06-04 16:24:16 +08:00
committed by GitHub
parent eb2701e0b2
commit 517811449e
5 changed files with 38 additions and 11 deletions

View File

@@ -16,6 +16,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
#
import os
import pytest
import torch
from vllm import LLM, SamplingParams
@@ -24,7 +26,11 @@ from vllm.utils import GiB_bytes
from tests.utils import fork_new_process_for_each_test
from vllm_ascend.device_allocator.camem import CaMemAllocator
if os.getenv("VLLM_USE_V1") == "1":
pytest.skip("Skip in vllm v1", allow_module_level=True)
@fork_new_process_for_each_test
def test_basic_camem():
# some tensors from default memory pool
shape = (1024, 1024)
@@ -57,7 +63,6 @@ def test_basic_camem():
assert torch.allclose(output, torch.ones_like(output) * 3)
@pytest.mark.skipif(True, reason="test failed, should be fixed later")
@fork_new_process_for_each_test
def test_end_to_end():
free, total = torch.npu.mem_get_info()