Skip to content

Commit 5afb010

Browse files
committed
add some more module-owners/modules
Signed-off-by: Venky Ganesh <[email protected]>
1 parent ce08e13 commit 5afb010

File tree

3 files changed

+45
-5
lines changed

3 files changed

+45
-5
lines changed

.github/module-paths.json

Lines changed: 21 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,25 @@
66
"benchmarks/": "Performance",
77
"examples/disaggregated/": "Disaggregated Serving",
88
"docs/": "Documentation",
9-
"docker/": "Installation"
9+
"docker/": "Installation",
10+
".github/": "CI/CD",
11+
"jenkins/": "CI/CD",
12+
"tensorrt_llm/_torch/": "Torch Framework",
13+
"tensorrt_llm/_torch/attention_backend/": "Torch Attention Backend",
14+
"tensorrt_llm/_torch/auto_deploy/": "Torch AutoDeploy",
15+
"tensorrt_llm/_torch/compilation/": "Torch Compilation",
16+
"tensorrt_llm/_torch/custom_ops/": "Torch Custom Ops",
17+
"tensorrt_llm/_torch/distributed/": "Torch Distributed",
18+
"tensorrt_llm/_torch/pyexecutor/": "Torch PyExecutor",
19+
"tensorrt_llm/_torch/speculative/": "Torch Speculative",
20+
"tensorrt_llm/autotuner.py": "Autotuner",
21+
"tensorrt_llm/pipeline_interface.py": "Pipeline Interface",
22+
"tensorrt_llm/_torch/models/": "Torch Models",
23+
"tensorrt_llm/_torch/models/modeling_deepseekv3.py": "Torch Models DeepSeekV3",
24+
"tensorrt_llm/_torch/models/modeling_llama.py": "Torch Models Llama",
25+
"tensorrt_llm/_torch/modules/": "Torch Modules",
26+
"tensorrt_llm/_torch/modules/attention.py": "Torch Modules Attention",
27+
"tensorrt_llm/_torch/modules/fused_moe.py": "Torch Modules Fused MOE",
28+
"tests/unittest/_torch/": "Torch Tests",
29+
"examples/pytorch/": "PyTorch Examples"
1030
}

.github/scripts/assign_reviewers.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import argparse
22
import json
33
import os
4+
import random
45
import subprocess
56
from pathlib import Path
67

@@ -66,8 +67,8 @@ def main() -> None:
6667
modules = map_modules(changed_files, module_paths)
6768
reviewers = gather_reviewers(modules, module_owners, pr_author=pr_author)
6869

69-
if reviewer_limit:
70-
reviewers = reviewers[:reviewer_limit]
70+
if reviewer_limit and len(reviewers) > reviewer_limit:
71+
reviewers = random.sample(reviewers, reviewer_limit)
7172

7273
if reviewers:
7374
cmd = ["gh", "pr", "edit", pr_number]

.github/workflows/module-owners.json

Lines changed: 21 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,10 +7,29 @@
77
"Speculative Decoding":["yweng0828", "nekorobov", "lfr-0531"],
88
"Customized Kernels":["lowsfer", "PerkzZheng", "jdemouth-nvidia"],
99
"Performance": ["kaiyux", "jiahanc", "hypdeb"],
10-
"Lora/P-tuning":["byshiue", "Naveassaf"],
10+
"Lora/P-tuning":["byshiue", "shaharmor98"],
1111
"Disaggregated Serving":["Shixiaowei02", "joyang-nv", "chuangz0", "schetlur-nv"],
1212
"Documentation":["nv-guomingz"],
1313
"Sampling": ["dcampora", "lfr-0531", "Naveassaf", "syuoni", "yweng0828"],
1414
"Memory": ["litaotju", "peaceh-nv"],
15-
"Installation": ["hchings", "Superjomn", "nv-guomingz", "QiJune"]
15+
"Installation": ["hchings", "Superjomn", "nv-guomingz", "QiJune"],
16+
"CI/CD": ["chzblych", "syuoni"],
17+
"Torch Framework": ["QiJune", "hlu1"],
18+
"Torch Attention Backend": ["yuxianq", "hlu1"],
19+
"Torch AutoDeploy": ["lucaslie", "suyoggupta"],
20+
"Torch Compilation": ["litaotju", "yizhang-nv", "liji-nv"],
21+
"Torch Custom Ops": ["yizhang-nv"],
22+
"Torch Distributed": ["yilin-void", "yuxianq", "hyukn", "yizhang-nv", "hlu1"],
23+
"Torch PyExecutor": ["dongxuy04", "funatiq", "dcampora", "HuiGao-NV"],
24+
"Torch Speculative": ["lfr-0531", "mikeiovine"],
25+
"Autotuner": ["hyukn", "litaotju"],
26+
"Pipeline Interface": ["amukkara", "chang-l"],
27+
"Torch Models": ["QiJune", "hlu1"],
28+
"Torch Models DeepSeekV3": ["hlu1", "zongfeijing"],
29+
"Torch Models Llama": ["chang-l", "mikeiovine"],
30+
"Torch Modules": ["QiJune", "hlu1"],
31+
"Torch Modules Attention": ["yuxianq", "hlu1"],
32+
"Torch Modules Fused MOE": ["hlu1", "dongxuy04", "zongfeijing", "HuiGao-NV"],
33+
"Torch Tests": ["QiJune", "hlu1"],
34+
"PyTorch Examples": ["QiJune", "hlu1"]
1635
}

0 commit comments

Comments
 (0)