Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
178 commits
Select commit Hold shift + click to select a range
2504aa5
test: add recursive updating pytorch config and change MOE backend fo…
ruodil Jul 15, 2025
2a147c4
test: add llama_v3.3_70b_cases in perf test (#6035)
ruodil Jul 15, 2025
9e871ca
[infra] add more log on reuse-uploading (#6036)
niukuo Jul 15, 2025
ab1c547
fix: adjust window sizes of VSWA at torch backend (#5880)
jaedeok-nvidia Jul 15, 2025
9ebc3ab
[nvbugs/5385972][nvbugs/5387423][Fix] Minor fix for llava_next/llava_…
MinaHuai Jul 15, 2025
4a26bd6
Fix: pad DeepEP fp4 recv tensors if empty (#6048)
yuantailing Jul 15, 2025
e761231
[fix] Move NCCL group in all-gather and reduce-scatter OPs outside th…
jinyangyuan-nvidia Jul 15, 2025
0523f77
support TRTLLM_DEEP_EP_TOKEN_LIMIT to allow run deep-ep on memory-con…
ttyio Jul 15, 2025
7a1af1c
Cherry-pick https://github.com/NVIDIA/TensorRT-LLM/pull/5947 (#5989)
lfr-0531 Jul 15, 2025
9214ac6
test: Add regression tests for Gemma3 VLM (#6033)
brb-nv Jul 15, 2025
edab753
feat/add latency support for trtllm bench (#3730)
danielafrimi Jul 15, 2025
6a47cac
feat: Add support for Triton request cancellation (#5898)
achartier Jul 16, 2025
665b446
[fix] Fix Triton build (#6076)
Tabrizian Jul 16, 2025
8679a05
fix: Unable to load phi4-model with tp_size>1 (#5962)
Wanli-Jiang Jul 16, 2025
e51c541
chore: Bump version to 1.0.0rc4 (#6086)
yiqingy0 Jul 16, 2025
509dc7c
chroe: upgrade modelopt to 0.33 (#6058)
nv-guomingz Jul 16, 2025
385af53
[nvbug/5347489][nvbug/5388036] increase timeout in disagg worker test…
zhengd-nv Jul 16, 2025
38db4bc
feat: use session abstraction in data transceiver and cache formatter…
zhengd-nv Jul 16, 2025
ec3ebae
[TRTLLM-6471] Infra: Upgrade NIXL to 0.3.1 (#5991)
bo-nv Jul 16, 2025
f5f31be
feat: Add deepseek-lite tests for RTX pro 6000 (#5903)
peaceh-nv Jul 16, 2025
763012a
[nvbug/5359218][tests] add test llm api test case on lookahead with c…
crazydemo Jul 16, 2025
7568deb
[nvbug/5387226] chore: add propogation for trust_remote_code to AutoC…
Superjomn Jul 16, 2025
dda91b5
tests: add QA test cases (#5959)
crazydemo Jul 16, 2025
10349b5
fix: Add $HOME/.local/bin to PATH when running docker in local user m…
MartinMarciniszyn Jul 16, 2025
a02606a
[TRTLLM-5530][BREAKING CHANGE] refactor: unify KvCacheConfig in LLM c…
Superjomn Jul 16, 2025
0552a02
BlockManager copy constructor fix (#5982)
tshmilnvidia Jul 16, 2025
8ef8e73
update spec_dec (#6079)
qsang-nv Jul 16, 2025
fc2347e
chore: Cleanup disable_fp4_allgather. (#6006)
bobboli Jul 16, 2025
e42f5a9
infra: [TRTLLM-5879] Spilt single GPU test and multi GPU test into 2 …
ZhanruiSunCh Jul 16, 2025
e30d7be
[Infra] - Waive failed cases in post-merge on main (#6096)
EmmaQiaoCh Jul 16, 2025
301b78b
Add documentation for eagle3+disagg+dynamo (#6072)
Tabrizian Jul 16, 2025
9354114
fix: Update trtllm args issues with extra nested config (#5996)
Wanli-Jiang Jul 16, 2025
e0836f9
[TRTLLM-5493] Add core infrastructure to enable loading of custom che…
shaharmor98 Jul 16, 2025
fa34cb7
[refactor] Clean up drafter/resource manager creation logic (#5805)
mikeiovine Jul 16, 2025
e09e409
Fix: Enhance ModelConfig for kv cache size calculations (#5868)
qixiang-99 Jul 16, 2025
2d2b8ba
feat: TRTLLM-5574 Add phi-4-multimodal pytorch-backend support (#5644)
Wanli-Jiang Jul 16, 2025
28385f6
[TRTLLM-6070] docs: Add initial documentation for trtllm-bench CLI. (…
FrankD412 Jul 17, 2025
fe070a0
test: Update Llama4 Scout FP4 & FP8 accuracy tests (#5901)
chenfeiz0326 Jul 17, 2025
6e1aee6
[fix] Performance Optimization for MNNVL TwoShot Kernel (#5934)
timlee0212 Jul 17, 2025
4c364b9
infra: fix SBSA test stage (#6113)
ZhanruiSunCh Jul 17, 2025
7e033c3
Feat: Add vectorized loading for finalize kernel in MoE Trtllm backen…
ChristinaZ Jul 17, 2025
d4d21a1
[fix] Release slots with spec decode + disagg (#5975) (#6032)
Tabrizian Jul 17, 2025
48daa18
[None][infra] Set up the initial config for CodeRabbit (#6128)
chzblych Jul 17, 2025
e821c68
CI: update multi gpu test trigger file list (#6131)
QiJune Jul 17, 2025
8c1c9ef
fix: convert venv_prefix to str before comparison with base_prefix (#…
dc3671 Jul 17, 2025
1cc4949
[Infra] - Add wiave list for pytest when using slurm (#6130)
EmmaQiaoCh Jul 17, 2025
44c70c8
chore:[BREAKING CHANGE] use cacheTransceiverConfig as knobs for disag…
chuangz0 Jul 17, 2025
21efb50
[TRTLLM-6406] feat: Enable guided decoding with overlap scheduler (#6…
syuoni Jul 17, 2025
de60ae4
chores: unwaive a few tests for v1.0 (#6107)
hchings Jul 17, 2025
9b45499
test: update max_beam_width to 1 due to torchsampler changes. (#6101)
nv-guomingz Jul 17, 2025
a718486
fix: Fix DeepSeek R1 CI (#6129)
yizhang-nv Jul 17, 2025
9518e14
test: fix PytestUnknownMarkWarning: Unknown pytest.mark.timeout (#6115)
StanleySun639 Jul 17, 2025
58d22a7
[TRTLLM-6352][feat] Migrate EAGLE3 and draft/target speculation to Dr…
ziyixiong-nv Jul 17, 2025
5bff317
feat: nanobind bindings (#5961)
Linda-Stadter Jul 17, 2025
d71c6fe
[fix] Update jenkins container images (#6094)
ixlmar Jul 17, 2025
10dbf4f
[fix] Remove duplicated KVCache transmission check (#6022)
Tabrizian Jul 17, 2025
8480c12
[fix] Fix Mistral3VLM weight-loading & enable in pre-merge (#6105)
2ez4bz Jul 17, 2025
161490f
[fix] Fixes KV Cache overrides in trtllm-bench (#6103)
FrankD412 Jul 17, 2025
2c90203
Refactor KVCacheManager: Simplify token availability calculation and …
qixiang-99 Jul 17, 2025
ae28b3a
feat: Add support for benchmarking individual gemms in MOE benchmark …
djns99 Jul 17, 2025
b75e53a
Revert "feat: nanobind bindings (#5961)" (#6160)
Tabrizian Jul 18, 2025
0155e7a
[TRTLLM-6368] Update deepep dispatch API (#6037)
yifeizhang-c Jul 18, 2025
200ea9e
fix TMA error with GEMM+AR on TP=2 (#6075)
xavier-nvidia Jul 18, 2025
992b273
[https://nvbugs/5387375] fix(scaffolding): fix scaffolding aime test …
dc3671 Jul 18, 2025
812243b
feat: add support for Modelopt fp8_pb_wo quantization scheme (#6106)
achartier Jul 18, 2025
c0e4165
fix single_disagg_test (#6166)
chuangz0 Jul 18, 2025
f321692
[TRTLLM-5179] - Update bot help messages (#5277)
yiqingy0 Jul 18, 2025
519a211
[None][infra] Update the allow list of CI trigger (#6168)
niukuo Jul 18, 2025
a95f31e
chore: add more log in FmhaDispatcher (#6170)
QiJune Jul 18, 2025
77acb4f
[Infra] - Waive failed tests in post-merge (#6176)
EmmaQiaoCh Jul 18, 2025
ec2b953
refactor: Enhanced handling of decoder requests and logits within the…
Funatiq Jul 18, 2025
44040ed
update broken link of PyTorchModelEngine in arch_overview (#6171)
leslie-fang25 Jul 18, 2025
9522cde
fix: NVBug 5385576 py_batch_idx issue (#6153)
hchings Jul 18, 2025
8454640
infra: fix single-GPU stage failed will not raise error (#6165)
ZhanruiSunCh Jul 18, 2025
fd6ce7f
[ci] Speedup beam search unit tests with fixtures for LLM (#5843)
stnie Jul 18, 2025
07e8813
feat: Remove padding in attention DP. (#6064)
bobboli Jul 18, 2025
2c6fa14
[TRTLLM-6471] Infra: unwaive nixl tests and some disagg-serve tests (…
bo-nv Jul 18, 2025
22d4a8c
enh: Add script to map tests <-> jenkins stages & vice-versa (#5177)
venkywonka Jul 18, 2025
28858c8
feat(eagle3):support qwen3 dense model (#5879)
xq25478 Jul 18, 2025
6d7874a
[nvbugs/5369799] fix: Update disaggregation handling in sampler (#5762)
stnie Jul 18, 2025
d475c97
[nvbugs/5354884][fix] Update beam search workspace estimation to new …
stnie Jul 18, 2025
d9a3530
[nvbug/5393888][nvbug/5393042] Always use `py_seq_slot` (#6147)
netanel-haber Jul 18, 2025
0388ff9
[https://nvbugs/5393961][fix] record kv-cache size in MLACacheFormatt…
bo-nv Jul 18, 2025
fc8b29c
[Issue 5927][fix] Avoid memory calls during broadcast for single GPU …
johncalesp Jul 18, 2025
152e2df
[Disaggregated] Add retry knobs and handling (#5808)
arekay Jul 18, 2025
82d3587
[refactor] Unify name of NGram speculative decoding (#5937)
wili-65535 Jul 19, 2025
66030ef
[TRTLLM-6452][feat]: Two-model engine KV cache reuse support (#6133)
ziyixiong-nv Jul 19, 2025
69e9f6d
[fix]: Skip prompt length checking for generation only requests (#6146)
LinPoly Jul 19, 2025
118307c
DeepEP LL support variable hidden size and tokens num (#6141)
yilin-void Jul 20, 2025
2e14c8f
[Fix][Chore][Qwen3] fix bug of using fp4 on sm120 (#6065)
byshiue Jul 20, 2025
943fd41
fix: Ensure mlx5 library is installed for deep_ep and remove deprecat…
MartinMarciniszyn Jul 20, 2025
98428f3
[TRTLLM-5826][feat] Support pytorch LoRA adapter eviction (#5616)
amitz-nv Jul 20, 2025
5300a99
W4A8 GEMM (#6005)
danielafrimi Jul 20, 2025
a433eba
enh: Lift expectation of single image per sample in Gemma3 VLM (#6195)
brb-nv Jul 21, 2025
6a3c9f8
test: add phi-4 multimodel and bielik-11b-v2.2 models for perf test (…
ruodil Jul 21, 2025
ca9bc57
fix: Flush stale `PlanParams` with custom attention mask (#6163)
brb-nv Jul 21, 2025
b4c7e8c
doc: remove cuda_graph_config: {} from doc since cuda_graph enabled b…
nv-guomingz Jul 21, 2025
88076ee
[fix] Fix can_use_alltoall in fused_moe_wide_ep.py (#6173)
jinyangyuan-nvidia Jul 21, 2025
e8c068b
[TRTLLM-5863][feat] Support Weight-Only-Quantization in PyTorch Workf…
Yuening-wa Jul 21, 2025
b46fd41
test: [CI] remove closed bugs (#6201)
xinhe-nv Jul 21, 2025
3efad2e
feat: nanobind bindings (#6185)
Linda-Stadter Jul 21, 2025
3cbc23f
infra: [TRTLLM-5250] Add sanity check stage for ngc-release images (B…
ZhanruiSunCh Jul 21, 2025
aea91b2
doc: add Deprecation Policy section (#5784)
QiJune Jul 21, 2025
3e0fb60
[TRTLLM-4279] feat: Multistream initial support for torch compile flo…
liji-nv Jul 21, 2025
e41507a
[Infra] - Waive failed cases on recent post-merge (#6212)
EmmaQiaoCh Jul 21, 2025
9832bef
[BREAKING CHANGE]: change default backend to PyTorch in trtllm-serve …
LinPoly Jul 21, 2025
f9b0a91
test: Enable GB200 torch compile multi gpu tests (#6145)
yizhang-nv Jul 21, 2025
d7f0b0a
[fix] Correct the returned value of has_spec_drafter (#6178)
ziyixiong-nv Jul 21, 2025
9645814
[chore] Clean up quickstart_advanced.py (#6021)
mikeiovine Jul 21, 2025
4a0951f
[Chore] Replace MODEL_CACHE_DIR with LLM_MODELS_ROOT and unwaive trit…
SimengLiu-nv Jul 21, 2025
7381f1d
[TRTLLM-5059][feat] Add KV cache reuse support for multimodal models …
chang-l Jul 21, 2025
ee45e0c
feat: Refactor the fetching request logic (#5786)
Shunkangz Jul 22, 2025
eb5cb5b
tests: add timeout_manager to tensorrt flow test cases (#5942)
crazydemo Jul 22, 2025
fddb7f1
feat: moe prepare support topk % 4 != 0 (#5742)
WeiHaocheng Jul 22, 2025
37d0b68
[fix] Fix flaky mistral E2E test (#6230)
2ez4bz Jul 22, 2025
db77d83
bug: [https://nvbugs/5368507] Fix test_generate_with_seed. (#6206)
bobboli Jul 22, 2025
537757e
fix: [nvbugs/5351130] Adjust DSV3-Lite tests free_gpu_memory_fraction…
bobboli Jul 10, 2025
f4f2176
chore: Port leftover 0.20 (#5907)
amirkl94 Jul 10, 2025
f194b65
fix [nvbug/5351244]: address remote mpi session submit (#5664)
Superjomn Jul 10, 2025
9d26b78
fix: [5328141] increase tolerance for test_fp8_block_scale_gemm (#5849)
nekorobov Jul 10, 2025
c669410
fix: fix index out of bounds error in spec decoding (#5954)
lfr-0531 Jul 14, 2025
eb7d0f8
[nvbugs/5368410][fix] Disable moe allreduce for multi node (#5918)
yizhang-nv Jul 14, 2025
34dd071
[TRTLLM-6495] doc: add disclaimer for 3rd party software installation…
nv-guomingz Jul 15, 2025
a03c680
add release notes for 0.21 release (#6049)
QiJune Jul 16, 2025
310bdd9
fix: Fix triton backend build [nvbug 5396469] (#6098)
pcastonguay Jul 16, 2025
24ce6b9
[Doc][Qwen3] update qwen3 into support-matrix (#6161)
byshiue Jul 18, 2025
48ddc3d
[fix]: Revert commit 388b491 (#6143)
LinPoly Jul 18, 2025
b85ab13
doc: add supported data modality and types on multimodal serve (#5988)
yechank-nvidia Jul 22, 2025
3e18ee5
chore: bump version to 1.0.0rc5 (#6252)
yiqingy0 Jul 22, 2025
3e1a0fb
[TRTLLM-6537][infra] extend multi-gpu tests related file list (#6139)
reasonsolo Jul 22, 2025
04f2d4b
test: update test list for RTX6KD (#6213)
StanleySun639 Jul 22, 2025
6007373
fix: bindings unit tests for nanobind (#6221)
Linda-Stadter Jul 22, 2025
ff99639
Add register_fake for finegrained_mixed_dtype_gemm torch_op (#6255)
danielafrimi Jul 22, 2025
b7c8a67
[Issue 6193] Fix gemma3vl weight loader (#6233)
johncalesp Jul 22, 2025
ab7434a
[feat] Enable TP and batching for PixtralVisionModel / Mistral3VLM (#…
2ez4bz Jul 22, 2025
ef4878d
set NVIDIA_IMEX_CHANNELS for dlcluster slurm job only (#6234)
yuanjingx87 Jul 22, 2025
5234502
[nvbug/5361223] doc: Update Llama4 deployment guide: update config & …
raayandhar Jul 22, 2025
41fb8aa
[AutoDeploy] merge feat/ad-2025-07-07 (#6196)
lucaslie Jul 22, 2025
bc2fb29
[nvbugs/5401261][fix] Fix Triton backend disaggregated serving suppor…
Tabrizian Jul 22, 2025
8ecdeee
[refactor] Simplification of Speculative decoding configs - Part 2 (#…
wili-65535 Jul 23, 2025
f08286c
doc: Refactor documents and examples of disaggregated serving and wid…
kaiyux Jul 23, 2025
9538c8d
Add basic Nemo Ckpt Lora Loading in pytorch flow (#6019)
venkywonka Jul 23, 2025
2193ad3
[https://nvbugs/5387771] fix deadlocks due to insufficient numSemapho…
PerkzZheng Jul 23, 2025
5636c67
fix: nvbug_5398806 (#6239)
hchings Jul 23, 2025
83c3ed1
chore: set default device to cpu on Multimodal models (#5994)
yechank-nvidia Jul 23, 2025
a8253b9
chore: remove duplicate should_stop_processing check (#6242)
QiJune Jul 23, 2025
fca13b8
hopper-style context MLA (#5713)
zhou-yuxin Jul 23, 2025
ed62a06
[nvbug/5322354] fix PD + MTP + overlap scheduler accuracy issue (#6136)
yweng0828 Jul 23, 2025
2b0fa24
test: [CI] Add failed cases into waives.txt (#6289)
xinhe-nv Jul 23, 2025
2486eb7
[TRTLLM-6651][feat] Enable Overlap scheduler + Beam Search in TRTLL…
stnie Jul 23, 2025
cb737a5
[Infra] - Skip failed cases (#6299)
EmmaQiaoCh Jul 23, 2025
cf4f4e8
[AutoDeploy] disable flaky MoE nvfp4 test (#6302)
lucaslie Jul 23, 2025
19696a6
[feat] Update .coderabbit.yaml with review settings and code guidelin…
venkywonka Jul 23, 2025
7740bfa
Waive tests (#6312)
Tabrizian Jul 24, 2025
82d03ca
[Infra] - Increase unittest execution time since some test exceeds 16…
EmmaQiaoCh Jul 24, 2025
5fceaa6
Revert "tests: add timeout_manager to tensorrt flow test cases (#5942…
Tabrizian Jul 24, 2025
31d3eff
doc: fix invalid links related with llm api example (#6317)
nv-guomingz Jul 24, 2025
428e340
chore: remove unused variables in pyexecutor (#6280)
QiJune Jul 24, 2025
a63a1ac
[TRTLLM-6444] Add some UCX trouble shooting docs and print UCX relate…
reasonsolo Jul 24, 2025
14d94a3
feat: Add non UB AR + Residual + Norm + Quant fusion (#6320)
liji-nv Jul 24, 2025
0ffcf9a
Update fmhaRunner.cpp to fix guardwords scan error (#6327)
zhou-yuxin Jul 24, 2025
f290108
tests: only get timeout value from pytest marker (#6287)
crazydemo Jul 24, 2025
0cc1f8c
[Infra] - Wiave failed tests in post-merge (#6331)
EmmaQiaoCh Jul 24, 2025
7b6aadc
[Fix][nvbug 5401163][nvbug 5404726][Qwen3] Fix bug of MoE on tp > 1 w…
byshiue Jul 24, 2025
62298bc
perf: customize cublastLt algo for Llamba 3.3 70B TP4 (#6315)
zhenhuaw-me Jul 24, 2025
706f421
[Fix] the bug in the trtllm-gen heurisitcf for MLA kernels. (#6284)
PerkzZheng Jul 24, 2025
ff72ca9
Improve TransferAgentTest.SyncMessage (#6250)
bo-nv Jul 24, 2025
0df758e
[TRTLLM-6650][feat] Enhance beam search support with CUDA graph integ…
stnie Jul 24, 2025
f8f5ba6
[fix] Update to remove popping of KV cache and other args. (#6310)
FrankD412 Jul 24, 2025
375f74e
[fix][nvbugs/5399355] Fix Lamport buffer clear issue for MNNVL TwoSho…
timlee0212 Jul 25, 2025
9a99e6d
fix: integration tests with nanobind (#6326)
Linda-Stadter Jul 25, 2025
0f2f11f
[TRTLLM-6453][feat] Support chunked prefill on spec decode 2 model (#…
mikeiovine Jul 25, 2025
2dcfa90
test: skip llama3.3 70b test on cg4 (#6293)
xinhe-nv Jul 25, 2025
d974198
[TRTLLM-5312] - Add bot run rules for triton tests (#4988)
yiqingy0 Jul 25, 2025
6268a60
tests: add test_chunked_prefill for llama4 (#5549)
xinhe-nv Jul 25, 2025
9d192e8
Merge current waive list with TOT waive list
yiqingy0 Jul 8, 2025
6b7c74f
fix for review
yiqingy0 Jul 22, 2025
27225ad
fix pre-commit check
yiqingy0 Jul 22, 2025
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
  •  
  •  
  •  
2 changes: 1 addition & 1 deletion .clangd
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ CompileFlags:
# Tweak the clangd parse settings for all files
CompileFlags:
Compiler: clang++
CompilationDatabase: .
CompilationDatabase: cpp/build
Add:
# report all errors
- "-ferror-limit=0"
Expand Down
40 changes: 40 additions & 0 deletions .coderabbit.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

# yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json
# https://docs.coderabbit.ai/getting-started/configure-coderabbit/
# In PR, comment "@coderabbitai configuration" to get the full config including defaults
language: "en-US"
reviews:
profile: chill
auto_title_placeholder: '@coderabbitai title'
auto_title_instructions: 'Should follow the format: "[fix/feat/doc/infra/...] \<summary of this PR\>". Keep it concise.'
commit_status: false
collapse_walkthrough: true
assess_linked_issues: true
related_issues: true
related_prs: true
suggested_labels: true
auto_apply_labels: true
suggested_reviewers: true
auto_assign_reviewers: true
poem: false
auto_review:
drafts: true
base_branches: ["main", "release/.+"]
knowledge_base:
code_guidelines:
enabled: true
filePatterns: ["**/CODING_GUIDELINES.md"]
38 changes: 28 additions & 10 deletions .github/pull_request_template.md
Original file line number Diff line number Diff line change
@@ -1,17 +1,24 @@
@coderabbitai summary

# PR title

Please write the PR title by following template:
<!--
Please write the PR title by following this template:

[JIRA ticket link/nvbug link/github issue link][fix/feat/doc/infra/...] \<summary of this PR\>
[JIRA ticket/NVBugs ID/GitHub issue][fix/feat/doc/infra/...] \<summary of this PR\>

For example, assume I have a PR hope to support a new feature about cache manager of Jira TRTLLM-1000 ticket, it would be like
For example, assume I have a PR to support a new feature about cache manager for JIRA ticket TRTLLM-1000, it would be like:

[TRTLLM-1000][feat] Support a new feature about cache manager

Or I have a PR to fix a Llama3 accuracy issue:

[https://nvbugs/1234567][fix] Fix Llama3 accuracy issue
-->

## Description

<!--
Please explain the issue and the solution in short.
-->

## Test Coverage

Expand All @@ -31,29 +38,40 @@ See details below for each supported subcommand.

<details>

`run [--disable-fail-fast --skip-test --stage-list "A10-1, xxx" --gpu-type "A30, H100_PCIe" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage "H100_PCIe-[Post-Merge]-1, xxx"]`
`run [--reuse-test (optional)pipeline-id --disable-fail-fast --skip-test --stage-list "A10-PyTorch-1, xxx" --gpu-type "A30, H100_PCIe" --test-backend "pytorch, cpp" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx" --detailed-log --debug(experimental)]`

Launch build/test pipelines. All previously running jobs will be killed.

`--reuse-test (optional)pipeline-id ` *(OPTIONAL)* : Allow the new pipeline to reuse build artifacts and skip successful test stages from a specified pipeline or the last pipeline if no pipeline-id is indicated. If the Git commit ID has changed, this option will be always ignored. The DEFAULT behavior of the bot is to reuse build artifacts and successful test results from the last pipeline.

`--disable-reuse-test ` *(OPTIONAL)* : Explicitly prevent the pipeline from reusing build artifacts and skipping successful test stages from a previous pipeline. Ensure that all builds and tests are run regardless of previous successes.

`--disable-fail-fast ` *(OPTIONAL)* : Disable fail fast on build/tests/infra failures.

`--skip-test ` *(OPTIONAL)* : Skip all test stages, but still run build stages, package stages and sanity check stages. Note: Does **NOT** update GitHub check status.

`--stage-list "A10-1, xxx"` *(OPTIONAL)* : Only run the specified test stages. Examples: "A10-1, xxx". Note: Does **NOT** update GitHub check status.
`--stage-list "A10-PyTorch-1, xxx"` *(OPTIONAL)* : Only run the specified test stages. Examples: "A10-PyTorch-1, xxx". Note: Does **NOT** update GitHub check status.

`--gpu-type "A30, H100_PCIe"` *(OPTIONAL)* : Only run the test stages on the specified GPU types. Examples: "A30, H100_PCIe". Note: Does **NOT** update GitHub check status.

`--test-backend "pytorch, cpp"` *(OPTIONAL)* : Skip test stages which don't match the specified backends. Only support [pytorch, cpp, tensorrt, triton]. Examples: "pytorch, cpp" (does not run test stages with tensorrt or triton backend). Note: Does **NOT** update GitHub pipeline status.

`--only-multi-gpu-test ` *(OPTIONAL)* : Only run the multi-GPU tests. Note: Does **NOT** update GitHub check status.

`--disable-multi-gpu-test ` *(OPTIONAL)* : Disable the multi-GPU tests. Note: Does **NOT** update GitHub check status.

`--add-multi-gpu-test ` *(OPTIONAL)* : Force run the multi-GPU tests. Will also run L0 pre-merge pipeline.
`--add-multi-gpu-test ` *(OPTIONAL)* : Force run the multi-GPU tests in addition to running L0 pre-merge pipeline.

`--post-merge ` *(OPTIONAL)* : Run the L0 post-merge pipeline instead of the ordinary L0 pre-merge pipeline.

`--extra-stage "H100_PCIe-[Post-Merge]-1, xxx"` *(OPTIONAL)* : Run the ordinary L0 pre-merge pipeline and specified test stages. Examples: --extra-stage "H100_PCIe-[Post-Merge]-1, xxx".
`--extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx"` *(OPTIONAL)* : Run the ordinary L0 pre-merge pipeline and specified test stages. Examples: --extra-stage "H100_PCIe-TensorRT-Post-Merge-1, xxx".

`--detailed-log ` *(OPTIONAL)* : Enable flushing out all logs to the Jenkins console. This will significantly increase the log volume and may slow down the job.

`--debug ` *(OPTIONAL)* : **Experimental feature**. Enable access to the CI container for debugging purpose. Note: Specify exactly one stage in the `stage-list` parameter to access the appropriate container environment. Note: Does **NOT** update GitHub check status.

For guidance on mapping tests to stage names, see `docs/source/reference/ci-overview.md`.
For guidance on mapping tests to stage names, see `docs/source/reference/ci-overview.md`
and the `scripts/test_to_stage_mapping.py` helper.

### kill

Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/blossom-ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ jobs:
startsWith(github.event.comment.body, '/bot skip --comment') ||
startsWith(github.event.comment.body, '/bot reuse-pipeline') ||
startsWith(github.event.comment.body, '/bot kill')) && contains(
fromJson('["byshiue","chuangz0","funatiq","hypdeb","jdemouth-nvidia","joyang-nv","lowsfer","Tabrizian","yweng0828","Shixiaowei02","MartinMarciniszyn","schetlur-nv","dcampora","pcastonguay","Naveassaf","lfr-0531","nekorobov","PerkzZheng","kaiyux","nv-guomingz","LinPoly","thorjohnsen","jiahanc","latency1024","tburt-nv","zeroepoch","chzblych","niukuo","ZhanruiSunCh","EmmaQiaoCh","yiqingy0","achartier","suyoggupta","amukkara","mk-nvidia","QiJune","lucaslie","davidmlw","hlu1","nvzhou","syuoni","NVGaryJi","symphonylyh","hello-11","zongfeijing","Jackch-NV","jinyangyuan-nvidia","LarryXFly","crazydemo","jaedeok-nvidia","wm2012011492","rosenrodt","zhuoyao1012","xinhe-nv","Yuening-wa","Shunkangz","zhengd-nv","yibinl-nvidia","StanleySun639","KingsleyLiu-NV","kxdc","yingcanw","BestJuly","ChristinaZ","bobboli","xueweilnvidia","kunlunl","cherichy","lucifer1004","Autumn1998","litaotju","peaceh-nv","liji-nv","SimengLiu-nv","yuxianq","yechank-nvidia","vallis-neria","DylanChen-NV","Tracin","zhhuang-nv","ISEEKYAN","xupinjie","tongyuantongyu","laikhtewari","zhuolingwang","dominicshanshan","jershi425","shifangx","StudyingShao","Superjomn","dongjiyingdjy","guangyunh-nv","wili-65535","tiffany940107","DanBlanaru","mikeiovine","djns99","ruodil","xiaoweiw-nv","xuwchen","bashimao","yizhang-nv","hyukn","nvpohanh","yuki-666","juney-nvidia","barry-delaney","Kefeng-Duan","MinaHuai","yilin-void","jhaotingc","jmydurant","katec846","CarstyYou","Njuapp","Jie-Fang","nvbrantz","inocsin","ruoqianguo","chenfeiz0326","ming-wei","eopXD","longlee0622","dongfengy","georgeliu95","evezhier","rakib-hasan","shangz-ai","JyChang012","wangsiping1997","yuanjings-nvda","tomeras91","roikoren755","amirkl94","shaharmor98","danielafrimi","amitz-nv","hijkzzz","rzilberstein-nvidia","dc3671","hchings","yuhengxnv","dongxuy04","qiaoxj07","omera-nv","DomBrown","brb-nv","FrankD412","yuhsuan-t","Fridah-nv","a-mccarthy","HuiGao-NV","alexmsettle","meenchen","sugunav14","cjluo-nv","kyleliang-nv","chang-l","WeiHaocheng","qixiang-99","BatshevaBlack","ebarilanM","xmchen1987","lingjiew","heyuhhh","netanel-haber","jiefangz-nv","wyw1267","yunruis","sklevtsov-nvidia","jgangani","pamelap-nvidia","ixlmar","GalSha","Dido0o0","rabiel","nvzhihanj","milesial","fzmu727","zackyoray","RoeyAzran1992","viraatc","v-shobhit","yuanjingx87","uchihatmtkinu","nvrohanv","vegaluisjose","qsang-nv","ChunhuanLin","timlee0212","venkywonka","zbpatel","tijyojwad","shyeh25","zihaok","nv-yilinf","ttyio","farazkh80","yuantailing","JennyLiu-nv","moraxu","IzzyPutterman","nvchenghaoz","nvxuanyuc","poweiw","stnie","zhanga5","nzmora-nvidia","greg-kwasniewski1","linda-stadter","Tom-Zheng","vanshilshah97","ixlmar","MatthiasKohl","Wanli-Jiang", "arekay", "davidclark-nv", "2ez4bz", "tcherckez-nvidia", "MrGeva", "galagam", "limin2021", "dhansen-nvidia","talorabr","kanghui0204","wu6u3tw","hvagadia","xavier-nvidia","raayandhar","dbari","nvjullin","elvischenv","zhenhuaw-me","weireweire","yifeizhang-c","jiaganc","ziyixiong-nv","FelixXidddd","JunyiXu-nv","bo-nv","zerollzeng","RayenTian","ameynaik-hub"]'),
fromJson('["byshiue","chuangz0","funatiq","hypdeb","jdemouth-nvidia","joyang-nv","lowsfer","Tabrizian","yweng0828","Shixiaowei02","MartinMarciniszyn","schetlur-nv","dcampora","pcastonguay","Naveassaf","lfr-0531","nekorobov","PerkzZheng","kaiyux","nv-guomingz","LinPoly","thorjohnsen","jiahanc","latency1024","tburt-nv","zeroepoch","chzblych","niukuo","ZhanruiSunCh","EmmaQiaoCh","yiqingy0","achartier","suyoggupta","amukkara","mk-nvidia","QiJune","lucaslie","davidmlw","hlu1","nvzhou","syuoni","NVGaryJi","symphonylyh","hello-11","zongfeijing","Jackch-NV","jinyangyuan-nvidia","LarryXFly","crazydemo","jaedeok-nvidia","wm2012011492","rosenrodt","zhuoyao1012","xinhe-nv","Yuening-wa","Shunkangz","zhengd-nv","yibinl-nvidia","StanleySun639","KingsleyLiu-NV","kxdc","yingcanw","BestJuly","ChristinaZ","bobboli","xueweilnvidia","kunlunl","cherichy","lucifer1004","Autumn1998","litaotju","peaceh-nv","liji-nv","SimengLiu-nv","yuxianq","yechank-nvidia","vallis-neria","DylanChen-NV","Tracin","zhhuang-nv","ISEEKYAN","xupinjie","tongyuantongyu","laikhtewari","zhuolingwang","dominicshanshan","jershi425","shifangx","StudyingShao","Superjomn","dongjiyingdjy","guangyunh-nv","wili-65535","tiffany940107","DanBlanaru","mikeiovine","djns99","ruodil","xiaoweiw-nv","xuwchen","bashimao","yizhang-nv","hyukn","nvpohanh","yuki-666","juney-nvidia","barry-delaney","Kefeng-Duan","MinaHuai","yilin-void","jhaotingc","jmydurant","katec846","CarstyYou","Njuapp","Jie-Fang","nvbrantz","inocsin","ruoqianguo","chenfeiz0326","ming-wei","eopXD","longlee0622","dongfengy","georgeliu95","evezhier","rakib-hasan","shangz-ai","JyChang012","wangsiping1997","yuanjings-nvda","tomeras91","roikoren755","amirkl94","shaharmor98","danielafrimi","amitz-nv","hijkzzz","rzilberstein-nvidia","dc3671","hchings","yuhengxnv","dongxuy04","qiaoxj07","omera-nv","DomBrown","brb-nv","FrankD412","yuhsuan-t","Fridah-nv","a-mccarthy","HuiGao-NV","alexmsettle","meenchen","sugunav14","cjluo-nv","kyleliang-nv","chang-l","WeiHaocheng","qixiang-99","BatshevaBlack","ebarilanM","xmchen1987","lingjiew","heyuhhh","netanel-haber","jiefangz-nv","wyw1267","yunruis","sklevtsov-nvidia","jgangani","pamelap-nvidia","ixlmar","GalSha","Dido0o0","rabiel","nvzhihanj","milesial","fzmu727","zackyoray","RoeyAzran1992","viraatc","v-shobhit","yuanjingx87","uchihatmtkinu","nvrohanv","vegaluisjose","qsang-nv","ChunhuanLin","timlee0212","venkywonka","zbpatel","tijyojwad","shyeh25","zihaok","nv-yilinf","ttyio","farazkh80","yuantailing","JennyLiu-nv","moraxu","IzzyPutterman","nvchenghaoz","nvxuanyuc","poweiw","stnie","zhanga5","nzmora-nvidia","greg-kwasniewski1","linda-stadter","Tom-Zheng","vanshilshah97","ixlmar","MatthiasKohl","Wanli-Jiang", "arekay", "davidclark-nv", "2ez4bz", "tcherckez-nvidia", "MrGeva", "galagam", "limin2021", "dhansen-nvidia","talorabr","kanghui0204","wu6u3tw","hvagadia","xavier-nvidia","raayandhar","dbari","nvjullin","elvischenv","zhenhuaw-me","weireweire","yifeizhang-c","jiaganc","ziyixiong-nv","FelixXidddd","JunyiXu-nv","bo-nv","zerollzeng","RayenTian","ameynaik-hub","raymochen","shuyixiong","johncalesp","leslie-fang25","reasonsolo","zhou-yuxin","vadiklyutiy","yali-arch","NVShreyas","h-guo18","pengbowang-nv"]'),
github.actor)
steps:
- name: Check if comment is issued by authorized person
Expand Down
13 changes: 9 additions & 4 deletions .github/workflows/bot-command.yml
Original file line number Diff line number Diff line change
Expand Up @@ -46,17 +46,22 @@ jobs:
"Run `/bot [-h|--help]` to print this help message.\n\n" +
"See details below for each supported subcommand.\n\n" +
"<details>\n\n" +
"`run [--disable-fail-fast --skip-test --stage-list \"A10-1, xxx\" --gpu-type \"A30, H100_PCIe\" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage \"H100_PCIe-[Post-Merge]-1, xxx\"]`\n\n" +
"`run [--reuse-test (optional)pipeline-id --disable-fail-fast --skip-test --stage-list \"A10-PyTorch-1, xxx\" --gpu-type \"A30, H100_PCIe\" --test-backend \"pytorch, cpp\" --add-multi-gpu-test --only-multi-gpu-test --disable-multi-gpu-test --post-merge --extra-stage \"H100_PCIe-TensorRT-Post-Merge-1, xxx\" --detailed-log --debug(experimental)]`\n\n" +
"Launch build/test pipelines. All previously running jobs will be killed.\n\n" +
"`--reuse-test (optional)pipeline-id ` *(OPTIONAL)* : Allow the new pipeline to reuse build artifacts and skip successful test stages from a specified pipeline or the last pipeline if no pipeline-id is indicated. If the Git commit ID has changed, this option will be always ignored. The DEFAULT behavior of the bot is to reuse build artifacts and successful test results from the last pipeline.\n\n" +
"`--disable-reuse-test ` *(OPTIONAL)* : Explicitly prevent the pipeline from reusing build artifacts and skipping successful test stages from a previous pipeline. Ensure that all builds and tests are run regardless of previous successes.\n\n" +
"`--disable-fail-fast ` *(OPTIONAL)* : Disable fail fast on build/tests/infra failures.\n\n" +
"`--skip-test ` *(OPTIONAL)* : Skip all test stages, but still run build stages, package stages and sanity check stages. Note: Does **NOT** update GitHub check status.\n\n" +
"`--stage-list \"A10-1, xxx\"` *(OPTIONAL)* : Only run the specified test stages. Examples: \"A10-1, xxx\". Note: Does **NOT** update GitHub check status.\n\n" +
"`--stage-list \"A10-PyTorch-1, xxx\"` *(OPTIONAL)* : Only run the specified test stages. Examples: \"A10-PyTorch-1, xxx\". Note: Does **NOT** update GitHub check status.\n\n" +
"`--gpu-type \"A30, H100_PCIe\"` *(OPTIONAL)* : Only run the test stages on the specified GPU types. Examples: \"A30, H100_PCIe\". Note: Does **NOT** update GitHub check status.\n\n" +
"`--test-backend \"pytorch, cpp\"` *(OPTIONAL)* : Skip test stages which don't match the specified backends. Only support [pytorch, cpp, tensorrt, triton]. Examples: \"pytorch, cpp\" (does not run test stages with tensorrt or triton backend). Note: Does **NOT** update GitHub pipeline status.\n\n" +
"`--only-multi-gpu-test ` *(OPTIONAL)* : Only run the multi-GPU tests. Note: Does **NOT** update GitHub check status.\n\n" +
"`--disable-multi-gpu-test ` *(OPTIONAL)* : Disable the multi-GPU tests. Note: Does **NOT** update GitHub check status.\n\n" +
"`--add-multi-gpu-test ` *(OPTIONAL)* : Force run the multi-GPU tests. Will also run L0 pre-merge pipeline.\n\n" +
"`--add-multi-gpu-test ` *(OPTIONAL)* : Force run the multi-GPU tests in addition to running L0 pre-merge pipeline.\n\n" +
"`--post-merge ` *(OPTIONAL)* : Run the L0 post-merge pipeline instead of the ordinary L0 pre-merge pipeline.\n\n" +
"`--extra-stage \"H100_PCIe-[Post-Merge]-1, xxx\"` *(OPTIONAL)* : Run the ordinary L0 pre-merge pipeline and specified test stages. Examples: --extra-stage \"H100_PCIe-[Post-Merge]-1, xxx\".\n\n" +
"`--extra-stage \"H100_PCIe-TensorRT-Post-Merge-1, xxx\"` *(OPTIONAL)* : Run the ordinary L0 pre-merge pipeline and specified test stages. Examples: --extra-stage \"H100_PCIe-TensorRT-Post-Merge-1, xxx\".\n\n" +
"`--detailed-log ` *(OPTIONAL)* : Enable flushing out all logs to the Jenkins console. This will significantly increase the log volume and may slow down the job.\n\n" +
"`--debug ` *(OPTIONAL)* : **Experimental feature**. Enable access to the CI container for debugging purpose. Note: Specify exactly one stage in the `stage-list` parameter to access the appropriate container environment. Note: Does **NOT** update GitHub check status.\n\n" +
"### kill\n\n" +
"`kill `\n\n" +
"Kill all running builds associated with pull request.\n\n" +
Expand Down
Loading