Skip to content

Commit 8903f93

Browse files
authored
update ci (#3953)
1 parent 1023a67 commit 8903f93

File tree

2 files changed

+14
-1
lines changed

2 files changed

+14
-1
lines changed

tests/cov_pytest.ini

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,4 +22,3 @@ addopts =
2222
--ignore=tests/operators/test_flash_mask_attn.py
2323
--ignore=tests/operators/test_w4afp8_gemm.py
2424
--ignore=tests/operators/test_tree_mask.py
25-
--ignore=tests/model_loader/test_common_model.py

tests/model_loader/utils.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
import pytest
2424

2525
TokensIdText = list[tuple[list[int], str]]
26+
FD_CACHE_QUEUE_PORT = int(os.getenv("FD_CACHE_QUEUE_PORT", 8234))
2627

2728

2829
def clear_logs():
@@ -117,6 +118,19 @@ def clean_ports(ports_to_clean: list[int]):
117118
"""
118119
Kill all processes occupying the ports listed in PORTS_TO_CLEAN.
119120
"""
121+
try:
122+
result = subprocess.run(
123+
f"ps -efww | grep {FD_CACHE_QUEUE_PORT} | grep -v grep", shell=True, capture_output=True, text=True
124+
)
125+
for line in result.stdout.strip().split("\n"):
126+
if not line:
127+
continue
128+
parts = line.split()
129+
pid = int(parts[1])
130+
print(f"Killing PID: {pid}")
131+
os.kill(pid, signal.SIGKILL)
132+
except Exception as e:
133+
print(f"Failed to kill cache manager process: {e}, {str(traceback.format_exc())}")
120134
for port in ports_to_clean:
121135
kill_process_on_port(port)
122136

0 commit comments

Comments
 (0)