mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00

Files

Yang Wang 00d7d6f123 [1/3][ghstack] [vllm ci build setup ]setup lumen_cli (#160043 )

# Description
set up torch_cli using argparses

## Details:
- add vllm placeholer in the cli
- add unittest for cli command

see Readme.md to see how to run the cli

Pull Request resolved: https://github.com/pytorch/pytorch/pull/160043
Approved by: https://github.com/huydhn

2025-08-14 05:51:45 +00:00

991 B

Raw Permalink Blame History

🔧 Lumen_cli

A Python CLI tool for building and testing PyTorch-based components, using a YAML configuration file for structured, repeatable workflows.

Features

Build
- external projects (e.g. vLLM)

📦 Installation

at the root of the pytorch repo

pip install -e .ci/lumen_cli

Run the cli tool

The cli tool must be used at root of pytorch repo, as example to run build external vllm:

python -m cli.run build external vllm

this will run the build steps with default behaviour for vllm project.

to see help messages, run

python3 -m cli.run --help

Add customized external build logics

To add a new external build, for instance, add a new external build logics:

create the build function in cli/lib folder
register your target and the main build function at EXTERNAL_BUILD_TARGET_DISPATCH in cli/build_cli/register_build.py
[optional] create your ci config file in .github/ci_configs/${EXTERNAL_PACKAGE_NAME}.yaml

991 B Raw Permalink Blame History