[None][infra] AutoDeploy: Add trtllm runner for standalone llm-c#15630
[None][infra] AutoDeploy: Add trtllm runner for standalone llm-c#15630bmarimuthu-nv wants to merge 3 commits into
Conversation
00dde4e to
750b291
Compare
Signed-off-by: Balamurugan Marimuthu <246387390+bmarimuthu-nv@users.noreply.github.com>
750b291 to
fd91b34
Compare
|
@CodeRabbit summary |
✅ Action performedSummary regeneration triggered. |
📝 WalkthroughWalkthroughThe standalone package generator now copies the ChangesStandalone package runner harness
Estimated code review effort🎯 2 (Simple) | ⏱️ ~10 minutes 🚥 Pre-merge checks | ✅ 4 | ❌ 1❌ Failed checks (1 warning)
✅ Passed checks (4 passed)
✏️ Tip: You can configure your own custom pre-merge checks in the settings. ✨ Finishing Touches🧪 Generate unit tests (beta)
Comment |
Signed-off-by: Balamurugan Marimuthu <246387390+bmarimuthu-nv@users.noreply.github.com>
There was a problem hiding this comment.
Actionable comments posted: 1
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.
Inline comments:
In `@examples/auto_deploy/llmc/create_standalone_package.py`:
- Around line 505-514: The _copy_runners() helper is silently skipping missing
required runner sources, which can produce an incomplete standalone package.
Update the logic that iterates over EXAMPLE_FILES and EXAMPLE_DIRS to fail fast
by raising an error as soon as an expected file or directory under
AD_EXAMPLES_SRC is absent, instead of continuing and returning a partial count.
Keep the existing copy behavior for present sources, but make the missing-source
case explicit in _copy_runners().
🪄 Autofix (Beta)
Fix all unresolved CodeRabbit comments on this PR:
- Push a commit to this branch (recommended)
- Create a new PR with the fixes
ℹ️ Review info
⚙️ Run configuration
Configuration used: Path: .coderabbit.yaml
Review profile: CHILL
Plan: Enterprise
Run ID: 7c455f11-03dc-478a-941c-d1d22946d530
📒 Files selected for processing (1)
examples/auto_deploy/llmc/create_standalone_package.py
| for src_name, dst_name in EXAMPLE_FILES.items(): | ||
| src = os.path.join(AD_EXAMPLES_SRC, src_name) | ||
| if os.path.isfile(src): | ||
| os.makedirs(runners_dst, exist_ok=True) | ||
| shutil.copy2(src, os.path.join(runners_dst, dst_name)) | ||
| count += 1 | ||
| for dname in EXAMPLE_DIRS: | ||
| src = os.path.join(AD_EXAMPLES_SRC, dname) | ||
| if os.path.isdir(src): | ||
| count += _copy_tree(src, os.path.join(runners_dst, dname)) |
There was a problem hiding this comment.
🎯 Functional Correctness | 🟠 Major | ⚡ Quick win
Fail fast when expected runner sources are missing (Line 507, Line 513).
_copy_runners() currently skips missing required sources and continues, which can silently generate a broken standalone package. Raise immediately when required runner files/directories are absent.
Proposed fix
def _copy_runners(output_dir: str) -> int:
@@
for src_name, dst_name in EXAMPLE_FILES.items():
src = os.path.join(AD_EXAMPLES_SRC, src_name)
- if os.path.isfile(src):
- os.makedirs(runners_dst, exist_ok=True)
- shutil.copy2(src, os.path.join(runners_dst, dst_name))
- count += 1
+ if not os.path.isfile(src):
+ raise FileNotFoundError(f"Missing required runner file: {src}")
+ os.makedirs(runners_dst, exist_ok=True)
+ shutil.copy2(src, os.path.join(runners_dst, dst_name))
+ count += 1
for dname in EXAMPLE_DIRS:
src = os.path.join(AD_EXAMPLES_SRC, dname)
- if os.path.isdir(src):
- count += _copy_tree(src, os.path.join(runners_dst, dname))
+ if not os.path.isdir(src):
+ raise FileNotFoundError(f"Missing required runner directory: {src}")
+ count += _copy_tree(src, os.path.join(runners_dst, dname))
return count📝 Committable suggestion
‼️ IMPORTANT
Carefully review the code before committing. Ensure that it accurately replaces the highlighted code, contains no missing lines, and has no issues with indentation. Thoroughly test & benchmark the code to ensure it meets the requirements.
| for src_name, dst_name in EXAMPLE_FILES.items(): | |
| src = os.path.join(AD_EXAMPLES_SRC, src_name) | |
| if os.path.isfile(src): | |
| os.makedirs(runners_dst, exist_ok=True) | |
| shutil.copy2(src, os.path.join(runners_dst, dst_name)) | |
| count += 1 | |
| for dname in EXAMPLE_DIRS: | |
| src = os.path.join(AD_EXAMPLES_SRC, dname) | |
| if os.path.isdir(src): | |
| count += _copy_tree(src, os.path.join(runners_dst, dname)) | |
| for src_name, dst_name in EXAMPLE_FILES.items(): | |
| src = os.path.join(AD_EXAMPLES_SRC, src_name) | |
| if not os.path.isfile(src): | |
| raise FileNotFoundError(f"Missing required runner file: {src}") | |
| os.makedirs(runners_dst, exist_ok=True) | |
| shutil.copy2(src, os.path.join(runners_dst, dst_name)) | |
| count += 1 | |
| for dname in EXAMPLE_DIRS: | |
| src = os.path.join(AD_EXAMPLES_SRC, dname) | |
| if not os.path.isdir(src): | |
| raise FileNotFoundError(f"Missing required runner directory: {src}") | |
| count += _copy_tree(src, os.path.join(runners_dst, dname)) |
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.
In `@examples/auto_deploy/llmc/create_standalone_package.py` around lines 505 -
514, The _copy_runners() helper is silently skipping missing required runner
sources, which can produce an incomplete standalone package. Update the logic
that iterates over EXAMPLE_FILES and EXAMPLE_DIRS to fail fast by raising an
error as soon as an expected file or directory under AD_EXAMPLES_SRC is absent,
instead of continuing and returning a partial count. Keep the existing copy
behavior for present sources, but make the missing-source case explicit in
_copy_runners().
|
/bot run |
|
PR_Github #55868 [ run ] triggered by Bot. Commit: |
Summary by CodeRabbit
Summary by CodeRabbit
New Features
runners/trtllmlocation, including an updated entrypoint name and run instruction.Bug Fixes
Description
Test Coverage
PR Checklist
Please review the following before submitting your PR:
PR description clearly explains what and why. If using CodeRabbit's summary, please make sure it makes sense.
PR Follows TRT-LLM CODING GUIDELINES to the best of your knowledge.
Test cases are provided for new code paths (see test instructions)
If PR introduces API changes, an appropriate PR label is added - either
api-compatibleorapi-breaking. Forapi-breaking, includeBREAKINGin the PR title.Any new dependencies have been scanned for license and vulnerabilities
CODEOWNERS updated if ownership changes
Documentation updated as needed
Update tava architecture diagram if there is a significant design change in PR.
The reviewers assigned automatically/manually are appropriate for the PR.
Please check this after reviewing the above items as appropriate for this PR.
GitHub Bot Help
To see a list of available CI bot commands, please comment
/bot help.