8030a11b96
project_7 docker build 已在竞赛平台验证成功。 完整搬运 wudixzy/competition stack: - qwen3_5.py 2615 行 (12 个 corex .so 调用) - patch_ops.sh 251 行 (set -eo pipefail + cd dirname) - 12 prebuilt corex .so (SHA256 verified) - 13 CUDA .cu 源码 + 11 build scripts - 9 vendor overrides (block/sampler/scheduler) - transformers-4.55.3 offline wheel - computility-run.yaml: 262144 max-model-len, BI100 env vars - Dockerfile 结构不变 (COPY qwen3_6_scripts + RUN patch_ops.sh)
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import importlib.util
|
|
import pathlib
|
|
import shlex
|
|
from typing import Iterable, Optional, Sequence, Tuple
|
|
|
|
|
|
def package_root(pkg: str) -> pathlib.Path:
|
|
spec = importlib.util.find_spec(pkg)
|
|
if spec is None:
|
|
raise RuntimeError(f"package not found: {pkg}")
|
|
if not spec.submodule_search_locations:
|
|
raise RuntimeError(f"package has no package root: {pkg}")
|
|
return pathlib.Path(next(iter(spec.submodule_search_locations))).resolve()
|
|
|
|
|
|
def ensure_file(path: pathlib.Path) -> pathlib.Path:
|
|
if not path.is_file():
|
|
raise FileNotFoundError(str(path))
|
|
return path
|
|
|
|
|
|
def ensure_dir(path: pathlib.Path) -> pathlib.Path:
|
|
if not path.is_dir():
|
|
raise FileNotFoundError(str(path))
|
|
return path
|
|
|
|
|
|
def replace_once(path: pathlib.Path,
|
|
old: str,
|
|
new: str,
|
|
*,
|
|
required: bool = True,
|
|
already_contains: Optional[str] = None) -> bool:
|
|
path = ensure_file(path)
|
|
text = path.read_text()
|
|
marker = already_contains if already_contains is not None else new
|
|
if marker in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
if old not in text:
|
|
msg = f"anchor not found in {path}: {old[:120]!r}"
|
|
if required:
|
|
raise RuntimeError(msg)
|
|
print(f"[warn] {msg}")
|
|
return False
|
|
path.write_text(text.replace(old, new, 1))
|
|
print(f"[ok] patched: {path}")
|
|
return True
|
|
|
|
|
|
def replace_one_of(path: pathlib.Path,
|
|
replacements: Sequence[Tuple[str, str]],
|
|
*,
|
|
required: bool = True,
|
|
already_contains: Optional[str] = None) -> bool:
|
|
path = ensure_file(path)
|
|
text = path.read_text()
|
|
if already_contains is not None and already_contains in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
for _, new in replacements:
|
|
if new in text:
|
|
print(f"[skip] already patched: {path}")
|
|
return False
|
|
for old, new in replacements:
|
|
if old in text:
|
|
path.write_text(text.replace(old, new, 1))
|
|
print(f"[ok] patched: {path}")
|
|
return True
|
|
anchors = ", ".join(repr(old[:80]) for old, _ in replacements)
|
|
msg = f"anchor not found in {path}; tried: {anchors}"
|
|
if required:
|
|
raise RuntimeError(msg)
|
|
print(f"[warn] {msg}")
|
|
return False
|
|
|
|
|
|
def shell_env_line(name: str, value: pathlib.Path) -> str:
|
|
return f"{name}={shlex.quote(str(value))}"
|