init v0.23.0

Signed-off-by: Sun Ruoxi <sunruoxi@4paradigm.com>

tests/e2e/nightly/single_node/ops/conftest.py

@@ -0,0 +1,61 @@
+import time
+from datetime import datetime
+
+import pytest
+
+from vllm_ascend.ops.triton.triton_utils import init_device_properties_triton
+from vllm_ascend.utils import enable_custom_op
+
+init_device_properties_triton()
+enable_custom_op()
+DURATION_THRESHOLD = 120
+SLOW_COUNT_LIMIT = 5
+
+
+_per_file_slow_cases = {}
+_current_file = None
+
+
+def pytest_runtest_setup(item):
+    item.start_time = time.time()
+
+
+def pytest_runtest_teardown(item, nextitem):
+    global _current_file
+
+    file_path = item.fspath
+    if not hasattr(item, "start_time"):
+        return
+
+    duration = time.time() - item.start_time
+
+    if file_path not in _per_file_slow_cases:
+        _per_file_slow_cases[file_path] = 0
+
+    if duration > DURATION_THRESHOLD:
+        _per_file_slow_cases[file_path] += 1
+        cnt = _per_file_slow_cases[file_path]
+        print(f" Detected that the test case took too long, ({cnt}/{SLOW_COUNT_LIMIT}):{duration:.2f}s")
+
+        if cnt >= SLOW_COUNT_LIMIT:
+            print(f"\n The number of timeout test cases  {file_path}   ≥{SLOW_COUNT_LIMIT}\n")
+            _current_file = file_path
+
+
+def pytest_runtest_call(item):
+    if _current_file == item.fspath:
+        print(f"CASE SKIP:{item.nodeid}")
+        pytest.skip("The use case takes too long.")
+
+
+@pytest.hookimpl(tryfirst=True, hookwrapper=True)
+def pytest_runtest_makereport(item, call):
+    """Hook to add timestamp to test reports"""
+    start_time = datetime.now().strftime("[%H:%M:%S]")
+
+    outcome = yield
+
+    report = outcome.get_result()
+
+    if report.when == "call":
+        print(f"{start_time}")