first commit

Signed-off-by: Venky Ganesh <[email protected]>

Author: venkywonka

.github/module-paths.json

@@ -0,0 +1,10 @@
+{
+    "cpp/": "Generic Runtime",
+    "triton_backend/": "Triton Backend",
+    "tensorrt_llm/_torch/peft/": "Lora/P-tuning",
+    "tensorrt_llm/": "LLM API/Workflow",
+    "benchmarks/": "Performance",
+    "examples/disaggregated/": "Disaggregated Serving",
+    "docs/": "Documentation",
+    "docker/": "Installation"
+}

.github/scripts/assign_reviewers.py

@@ -0,0 +1,86 @@
+import argparse
+import json
+import os
+import subprocess
+from pathlib import Path
+
+
+def run_git_diff(base: str, head: str) -> list[str]:
+    result = subprocess.run(
+        ["git", "diff", "--name-only", base, head],
+        capture_output=True,
+        text=True,
+        check=True,
+    )
+    return [line.strip() for line in result.stdout.splitlines() if line.strip()]
+
+
+def load_json(path: str):
+    with open(path, "r", encoding="utf-8") as f:
+        return json.load(f)
+
+
+def map_modules(changed_files: list[str], module_paths: dict[str,
+                                                             str]) -> set[str]:
+    modules: set[str] = set()
+    for file in changed_files:
+        for prefix, module in module_paths.items():
+            if file.startswith(prefix):
+                modules.add(module)
+                break
+    return modules
+
+
+def gather_reviewers(modules: set[str],
+                     module_owners: dict[str, list[str]],
+                     *,
+                     pr_author: str | None = None) -> list[str]:
+    reviewers: set[str] = set()
+    for module in modules:
+        reviewers.update(module_owners.get(module, []))
+
+    if pr_author:
+        reviewers.discard(pr_author)
+
+    return sorted(reviewers)
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(
+        description="Assign reviewers based on changed modules")
+    parser.add_argument("--dry-run",
+                        action="store_true",
+                        help="Print the gh command instead of executing")
+    args = parser.parse_args()
+
+    base_sha = os.environ["BASE_SHA"]
+    head_sha = os.environ["HEAD_SHA"]
+    pr_number = os.environ["PR_NUMBER"]
+    reviewer_limit = int(os.environ.get("REVIEWER_LIMIT", "0"))
+    pr_author = os.environ.get("PR_AUTHOR")
+
+    changed_files = run_git_diff(base_sha, head_sha)
+    module_paths = load_json(Path(".github") / "module-paths.json")
+    module_owners = load_json(Path(".github/workflows") / "module-owners.json")
+
+    modules = map_modules(changed_files, module_paths)
+    reviewers = gather_reviewers(modules, module_owners, pr_author=pr_author)
+
+    if reviewer_limit:
+        reviewers = reviewers[:reviewer_limit]
+
+    if reviewers:
+        cmd = ["gh", "pr", "edit", pr_number]
+        for reviewer in reviewers:
+            cmd.extend(["--add-reviewer", reviewer])
+        if args.dry_run:
+            print("Dry run:", " ".join(cmd))
+        else:
+            subprocess.run(cmd, check=True)
+
+    print(f"Changed modules: {sorted(modules)}")
+    print(f"Requested reviewers: {reviewers}")
+
+
+if __name__ == "__main__":
+    main()

.github/workflows/auto-assign-reviewers.yml

@@ -0,0 +1,23 @@
+name: Auto assign reviewers
+on:
+  pull_request:
+    types: [opened, synchronize, reopened]
+permissions:
+  pull-requests: write
+jobs:
+  assign:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: '3.12'
+      - name: Assign reviewers
+        env:
+          BASE_SHA: ${{ github.event.pull_request.base.sha }}
+          HEAD_SHA: ${{ github.event.pull_request.head.sha }}
+          PR_NUMBER: ${{ github.event.pull_request.number }}
+          PR_AUTHOR: ${{ github.event.pull_request.user.login }}
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+          REVIEWER_LIMIT: '3'
+        run: python3 .github/scripts/assign_reviewers.py

CONTRIBUTING.md

@@ -93,6 +93,13 @@ Developer workflow for code contributions is as follows:
 3. Once the code changes are staged on the fork and ready for review, a [Pull Request](https://help.github.com/en/articles/about-pull-requests) (PR) can be [requested](https://help.github.com/en/articles/creating-a-pull-request) to merge the changes from a branch of the fork into a selected branch of upstream. PRs should typically target the `main` branch.
   * Creation of a PR creation kicks off the code review process.
   * At least one TensorRT-LLM engineer will be assigned for the review. When the PR is under review, the label `Pending Review` will be added to the PR.
+  * Reviewers are automatically requested based on the modules affected in the PR. Module paths are defined in `.github/module-paths.json` and ownership in `.github/workflows/module-owners.json`.
+  * You can test the assignment script locally with the `--dry-run` flag:
+    ```bash
+    GH_TOKEN=<token> BASE_SHA=<base> HEAD_SHA=<head> PR_NUMBER=<pr> \
+      PR_AUTHOR=<username> \
+      python3 .github/scripts/assign_reviewers.py --dry-run
+    ```
   * If changes are requested, then the reviewer will add the label `Changes Requested` to the PR.
   * Once changes are approved, CI will be launched to validate the change. When CI passes, the reviewer will merge the PR.
   * If CI reports any failures, it's up to the requester to fix any CI failures before requesting another review.