README.md 1.5 KB
Newer Older
sunzhq2's avatar
sunzhq2 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
# AISBench benchmark 链接
- https://gitee.com/aisbench/benchmark

- git clone https://gitee.com/aisbench/benchmark.git

- AISBench benchmark安装:具体安装可以参考 AISBench 对应的README

# aisbench-tools工具
- monitor_gpu.sh 用于记录DCU的使用状态
- run.sh
    - 随机输入输出
        ```
        bash ./bench-test.sh --gpus 3 \
                              --model vllm_api_stream_chat \
                              --dataset synthetic_gen \
                              --model-name test_model \
                              --batch-size  "8 16 32" \
                              --input-len 512 \
                              --max-out-len 512 \
                              --ais-model /data2/models/qwen3-8B \
                              --model-path /data2/models/qwen3-8B \
                              --host-port 23456
        ```

    - 对应数据集
        ```
        bash ./bench-test.sh --gpus 3 \
                     --model vllm_api_stream_chat \
                     --dataset aime2025_gen \
                     --model-name test_model \
                     --batch-size  "16" \
                     --max-out-len 512 \
                     --ais-model /data2/models/qwen3-8B \
                     --model-path /data2/models/qwen3-8B \
                     --host-port 23456    
        ```

    - 仅跑eval指令
        ```
        ais_bench --models vllm_api_stream_chat \
            --datasets aime2025_gen --mode eval \
            --debug --reuse xxxxxx
        ```