Skip to content
Merged
Show file tree
Hide file tree
Changes from 14 commits
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
0a81a5c
[CI] pin ruff and make its configuration explicit
valarLip Jul 28, 2026
ad8db44
fix: accept PEP 604 unions in generate_schema
valarLip Jul 28, 2026
e40c1cb
style: apply ruff's mechanical autofixes (batch 1/N)
valarLip Jul 28, 2026
b23a643
style: modernize type annotations (batch 2/N)
valarLip Jul 28, 2026
ed8e227
style: finish the batch 2 residue that ruff could not autofix
valarLip Jul 28, 2026
464dd84
style: sort imports where ordering is not load-bearing (batch 3a/N)
valarLip Jul 28, 2026
a459af7
style: clear the pre-existing findings this PR pulled into the diff
valarLip Jul 28, 2026
da4ba79
style: apply every remaining ruff autofix (safe and unsafe)
valarLip Jul 28, 2026
9b0975b
style: mechanical fixes for the rules ruff cannot autofix
valarLip Jul 28, 2026
ffde60b
Merge origin/main into ruff/pin-and-configure
valarLip Jul 28, 2026
6f02284
fix: correct two misplaced noqa directives
valarLip Jul 28, 2026
b652709
style: mark the deliberate blanket catches with noqa (BLE001)
valarLip Jul 28, 2026
a15e9fb
style: bind loop variables in closures (B023)
valarLip Jul 28, 2026
64c4228
style: make the warmup group access explicit (B018)
valarLip Jul 28, 2026
2f3ce97
style: clear S110/B008/TRY002/SIM102/PLR0124/PLW0127/TRY004
valarLip Jul 28, 2026
266c97f
style: drop the fx.Stream(None) parameter defaults (B008)
valarLip Jul 28, 2026
596c9d0
style: clear F821, SIM115 and UP031
valarLip Jul 28, 2026
c9af09e
style: clear the last 27 findings -- repo is ruff-clean
valarLip Jul 28, 2026
4a68321
fix: restore the torch import that F401 removed from core.py's eval path
valarLip Jul 28, 2026
0d763d2
fix: make ck_tile_gemm_moe_2stages codegen set up sys.path before imp…
valarLip Jul 28, 2026
7d4518c
fix: bind Tuple explicitly in the check_args exec namespace
valarLip Jul 28, 2026
0a4acd2
Merge branch 'main' into ruff/pin-and-configure
valarLip Jul 28, 2026
e688cb5
fix: restore the paged_attn binding test_pa_mtp lost to F401
valarLip Jul 28, 2026
783dc22
fix(test): pad the mxfp4 GUI scale reference with the neutral e8m0 scale
valarLip Jul 28, 2026
092427c
fix(test): drop the non-contiguity assert in the strided grouped-topk…
valarLip Jul 28, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
Empty file modified .github/scripts/aiter_prebuild_summary.py
100644 → 100755
Empty file.
Empty file modified .github/scripts/collect_logs.py
100644 → 100755
Empty file.
19 changes: 9 additions & 10 deletions .github/scripts/compare_benchmark.py
100644 → 100755
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@
import csv
import sys
from pathlib import Path
from typing import Dict, Tuple

METRIC = "us"
KERNEL_COLS = ("kernelName1", "kernelName2")
Expand Down Expand Up @@ -108,8 +107,8 @@ def _natural_key(val: str):
return (1, val)


Row = Dict[str, str]
Key = Tuple[Tuple[str, str], ...]
Row = dict[str, str]
Key = tuple[tuple[str, str], ...]


def _normalize_row(raw: Row) -> Row:
Expand All @@ -121,7 +120,7 @@ def _normalize_row(raw: Row) -> Row:
}


def _read_csv_rows(path: Path) -> Tuple[list[Row], Tuple[str, ...]]:
def _read_csv_rows(path: Path) -> tuple[list[Row], tuple[str, ...]]:
"""Return (rows, fieldnames). Whitespace stripped from values."""
if not path.exists():
raise SystemExit(f"input csv not found: {path}")
Expand All @@ -136,7 +135,7 @@ def _read_csv_rows(path: Path) -> Tuple[list[Row], Tuple[str, ...]]:
return rows, tuple(reader.fieldnames)


def _key_cols(base_cols: Tuple[str, ...], cur_cols: Tuple[str, ...]) -> Tuple[str, ...]:
def _key_cols(base_cols: tuple[str, ...], cur_cols: tuple[str, ...]) -> tuple[str, ...]:
"""Stable key column order across baseline/current schema drift."""
cols = []
seen = set()
Expand All @@ -148,8 +147,8 @@ def _key_cols(base_cols: Tuple[str, ...], cur_cols: Tuple[str, ...]) -> Tuple[st
return tuple(cols)


def _index_rows(rows: list[Row], key_cols: Tuple[str, ...]) -> Dict[Key, Row]:
indexed: Dict[Key, Row] = {}
def _index_rows(rows: list[Row], key_cols: tuple[str, ...]) -> dict[Key, Row]:
indexed: dict[Key, Row] = {}
for row in rows:
key = tuple(sorted((c, row.get(c, "")) for c in key_cols))
indexed[key] = row
Expand All @@ -167,7 +166,7 @@ def _parse_us(raw: Row) -> float | None:


def _fmt_key_compact(
key: Key, key_cols_order: Tuple[str, ...], constants: Dict[str, str]
key: Key, key_cols_order: tuple[str, ...], constants: dict[str, str]
) -> str:
"""Format key showing only cols whose value is NOT in `constants`."""
d = dict(key)
Expand All @@ -178,7 +177,7 @@ def _fmt_key_compact(
return " ".join(parts) if parts else "(common)"


def _find_constants(keys: list[Key], key_cols_order: Tuple[str, ...]) -> Dict[str, str]:
def _find_constants(keys: list[Key], key_cols_order: tuple[str, ...]) -> dict[str, str]:
"""Return cols whose value is identical across all `keys`."""
if not keys:
return {}
Expand Down Expand Up @@ -304,7 +303,7 @@ def _entry_sort_key(e):
if c not in HIDE_DISPLAY_COLS:
display_cols.append(c)

def _cell_value(c: str, d: Dict[str, str]) -> str:
def _cell_value(c: str, d: dict[str, str]) -> str:
if c in DERIVED_TUPLE_COLS:
srcs = DERIVED_TUPLE_COLS[c]
return "(" + ", ".join(d.get(s, "") for s in srcs) + ")"
Expand Down
Empty file modified .github/scripts/generate_summary.py
100644 → 100755
Empty file.
2 changes: 1 addition & 1 deletion .github/scripts/list_jobs.py
100644 → 100755
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def main():
content = yaml.safe_load(file_obj) or {}

jobs_dict = content.get("jobs") or {}
job_ids = [job_id for job_id in jobs_dict.keys() if job_id not in excluded_jobs]
job_ids = [job_id for job_id in jobs_dict if job_id not in excluded_jobs]

display_names = []
for job_id in job_ids:
Expand Down
13 changes: 5 additions & 8 deletions .github/scripts/query_job_status.py
100644 → 100755
Original file line number Diff line number Diff line change
Expand Up @@ -190,8 +190,7 @@ def list_jobs_for_run(owner: str, repo: str, run_id: int, token: str):
jobs = payload.get("jobs", [])
if not jobs:
return
for job in jobs:
yield job
yield from jobs
page += 1


Expand All @@ -202,9 +201,7 @@ def job_name_matches(filter_name: str, actual_name: str):
return True
if actual_name.startswith(f"{filter_name} ("):
return True
if actual_name.startswith(f"{filter_name} / "):
return True
return False
return bool(actual_name.startswith(f"{filter_name} / "))


def row_to_dict(row: dict[str, Any]):
Expand Down Expand Up @@ -360,7 +357,7 @@ def select_primary_runner_label(labels: list[str]):
)
]
candidates = preferred or labels
return sorted(candidates, key=lambda value: (-len(value), value.lower()))[0]
return min(candidates, key=lambda value: (-len(value), value.lower()))


def get_runner_label(row: dict[str, Any]):
Expand Down Expand Up @@ -459,7 +456,7 @@ def analyze_concurrency(job_rows: list[dict[str, Any]], report_time: datetime):
durations: list[float] = []

for row in label_rows:
created = parse_time(row.get("created_at", ""))
parse_time(row.get("created_at", ""))
started = parse_time(row.get("started_at", ""))
completed = parse_time(row.get("completed_at", ""))

Expand Down Expand Up @@ -816,7 +813,7 @@ def main():

if rate_limited and rate_limit_reset_epoch:
reset_time = datetime.fromtimestamp(rate_limit_reset_epoch, timezone.utc)
print("")
print()
print(
f"> NOTE: Partial data due to GitHub API rate limit. "
f"Reset at {reset_time.isoformat()} (UTC)."
Expand Down
Loading
Loading