Skip to content

Commit 2426579

Browse files
ceci3li199959keennddyljiamingwang-mtCopilot
authored
[cherry-pick] cherry pick commit_id bbb766d cefbff6 af71c54 3729890 28219b5 cbf88e2 from main to 0.1.0 (#116)
### PR Category <!-- One of [Core | Vendor | OP | Tools | Others] --> Others ### PR Type <!-- One of [User Experience | New Features | Bug Fixes | Improvements | Performance | Breaking Change | Deprecations | Test Case | Docs | Others] --> Others ### Description <!-- Describe what this PR does and why. --> cherry pick commit_id bbb766d cefbff6 af71c54 3729890 28219b5 cbf88e2 from main to 0.1.0 ### Related Issues <!-- Link any related issues: Fixes #issue, Closes #issue, or Related to #issue --> ### Changes <!-- List the key changes made in this PR. --> ### Testing <!-- How has this change been tested? Include test commands, hardware used, etc. --> - ### Checklist - [ ] I have run the existing tests and they pass - [ ] I have added tests for my changes (if applicable) - [ ] I have updated the documentation (if applicable) --------- Co-authored-by: xiangbin <74005582+li199959@users.noreply.github.com> Co-authored-by: keennddyl <keennddy@foxmail.com> Co-authored-by: jiamingwang-mt <jiaming.wang@mthreads.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: keennddyl <keenndd@foxmail.com> Co-authored-by: hozier <hozier@mthreads.com> Co-authored-by: cyber-pioneer <116002591+cyber-pioneer@users.noreply.github.com> Co-authored-by: XMing <xmhubj@gmail.com> Co-authored-by: xin2an <yuxinan@perfxlab.com>
1 parent 906fa07 commit 2426579

52 files changed

Lines changed: 1750 additions & 124 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

.github/ISSUE_TEMPLATE/blank.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,4 +5,3 @@ title: ''
55
labels: ''
66
assignees: ''
77
---
8-

.github/configs/cuda.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ container_volumes:
2020

2121
# Container options (hardware-specific settings)
2222
container_options: >-
23+
--privileged
2324
--gpus all
2425
--shm-size=500g
2526
--hostname vllm-plugin-fl

.github/scripts/cuda/setup.sh

100755100644
File mode changed.

.github/workflows/_unit_test.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -83,8 +83,10 @@ jobs:
8383
8484
- name: Upload coverage report to FlagCICD
8585
if: inputs.upload_coverage && always()
86-
uses: flagos-ai/FlagOps/actions/post-pytest-report@v1
86+
uses: flagos-ai/FlagOps/actions/post-pytest-report@v2
87+
continue-on-error: true
8788
with:
88-
backend_url: 'http://10.1.4.167:30180/flagcicd-backend/metrics/'
89+
backend_url: 'http://flagcicd-inner.flagos.net:8000/metrics/'
90+
user_id: '000000000000000000'
8991
report_path: 'coverage-${{ inputs.platform }}.json'
9092
fail_on_error: 'false'

README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ In theory, vllm-plugin-FL can support all models available in vLLM, as long as n
1616
| MiniCPM-o 4.5 | Supported | [example](./examples/minicpm/) |
1717
| GLM-5 | Supported | [example](./examples/glm_5_offline_inference.py) |
1818
| Qwen3.5-35B-A3B | Supported | [example](./examples/glm_5_offline_inference.py) |
19+
| BAAI/bge-m3 | Supported | [implementation](./vllm_fl/models/bge_m3.py) |
1920

2021
### Supported Chips
2122

benchmarks/flagos_eval/README.md

Lines changed: 37 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ Evaluation toolkit for large language models with LM Eval and vLLM Benchmark.
66

77
### 1. Dependencies
88

9-
* **[lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness)**
9+
* **[lm-evaluation-harness](https://github.com/EleutherAI/lm-evaluation-harness)**
1010
```bash
1111
git clone https://github.com/EleutherAI/lm-evaluation-harness.git
1212
cd lm-evaluation-harness
@@ -39,6 +39,10 @@ cd benchmarks/flagos_eval
3939
4040
# Performance Benchmark
4141
./run_benchmark.sh /path/to/model/
42+
43+
# Mixed-Length Benchmark (Vendor Acceptance)
44+
./run_mixed_benchmark.sh /path/to/model/ # default seed=42
45+
./run_mixed_benchmark.sh /path/to/model/ 123 # custom seed
4246
```
4347

4448
## Evaluation Tasks
@@ -65,16 +69,41 @@ Output Files:
6569
- `bench_summary.csv` - Summary
6670
- `bench_results/*.json` - Detailed results
6771

72+
### Mixed-Length Benchmark
73+
74+
Runs throughput tests with mixed input lengths in shuffled order for vendor performance acceptance testing. Uses a seed to ensure reproducible shuffle results.
75+
76+
Configuration (1610 samples total):
77+
78+
| Input/Output | Samples |
79+
|--------------|---------|
80+
| 1k/1k | 500 |
81+
| 2k/1k | 300 |
82+
| 4k/1k | 300 |
83+
| 6k/1k | 300 |
84+
| 12k/1k | 200 |
85+
| 32k/1k | 10 |
86+
87+
Output Files:
88+
- `mixed_bench_summary.csv` - Summary
89+
- `mixed_bench_summary.json` - Summary (JSON)
90+
- `mixed_bench_results/*.json` - Detailed results
91+
6892
## Project Structure
6993

7094
```
7195
flagos_eval/
72-
├── run_eval.sh # LM evaluation script
73-
├── run_benchmark.sh # Performance benchmark script
74-
├── collect_eval_results.py # LM evaluation result collector
96+
├── run_eval.sh # LM evaluation script
97+
├── run_benchmark.sh # Performance benchmark script
98+
├── run_mixed_benchmark.sh # Mixed-length benchmark script
99+
├── collect_eval_results.py # LM evaluation result collector
75100
├── collect_benchmark_results.py # Performance benchmark result collector
76-
├── output/ # LM evaluation results
77-
├── bench_results/ # Performance benchmark results
78-
├── results_summary.csv # LM evaluation summary
79-
└── bench_summary.csv # Performance benchmark summary
101+
├── collect_mixed_benchmark_results.py # Mixed-length result collector
102+
├── output/ # LM evaluation results
103+
├── bench_results/ # Performance benchmark results
104+
├── mixed_bench_results/ # Mixed-length benchmark results
105+
├── results_summary.csv # LM evaluation summary
106+
├── bench_summary.csv # Performance benchmark summary
107+
├── mixed_bench_summary.csv # Mixed-length benchmark summary
108+
└── mixed_bench_summary.json # Mixed-length benchmark summary (JSON)
80109
```

0 commit comments

Comments
 (0)