README.md 827 Bytes
Newer Older
sunzhq2's avatar
sunzhq2 committed
1
2
3
4
5
6
7
# 镜像使用
```
harbor.sourcefind.cn:5443/dcu/admin/base/vllm:0.11.0-ubuntu22.04-dtk26.04-0130-py3.10-20260204
```
# rocblas-bench
```
rocblas-bench用于测试矩阵乘法,使用前先通过下面的命令配置可执行文件
sunzhq2's avatar
sunzhq2 committed
8

sunzhq2's avatar
sunzhq2 committed
9
10
11
12
13
ln -s ${ROCM_PATH}/lib/rocblas/benchmark_tool/rocblas-bench ${ROCM_PATH}/bin/rocblas-bench

chmod +x ${ROCM_PATH}/bin/rocblas-bench
```

sunzhq2's avatar
sunzhq2 committed
14
- bash run_rocbals_bench.sh
sunzhq2's avatar
sunzhq2 committed
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

# hipblaslt
```
hipblaslt-bench 用于测试W8A8,使用前先通过下面的命令配置可执行文件

ln -s ${ROCM_PATH}/lib/hipblaslt/benchmark_tool/hipblaslt-bench ${ROCM_PATH}/bin/hipblaslt-bench
chmod +x ${ROCM_PATH}/bin/hipblaslt-bench
```

```
bash run_rocbals_bench.sh
```

# torch_gemm 测试
- run_gemm_benchmarks_torch.sh 和 gemm_benchmark.py 是用 torch 实现的gemm测试
- bash run_gemm_benchmarks_torch.sh