Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tsoc
hg-misc-tools
Commits
b52f967e
Commit
b52f967e
authored
Jan 30, 2026
by
one
Browse files
Update README with additional GEMV benchmark commands and parameter details
parent
57774b4a
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
12 additions
and
6 deletions
+12
-6
gemv/README.md
gemv/README.md
+12
-6
No files found.
gemv/README.md
View file @
b52f967e
...
...
@@ -4,10 +4,10 @@ GEMV Benchmarks
模仿 GEMM 接口的 GEMV,即 N=1,实现 BF16 版本。这些矩阵形状来自于 Evo2 推理过程。
计算公式:y = alpha
* A^T *
x + beta
*
y
M: 输出维度,例如 11264
K: 归约维度,例如 4096
N: 始终为 1
beta: 为 0 或 1
-
M: 输出维度,例如 11264
-
K: 归约维度,例如 4096
-
N: 始终为 1
-
beta: 为 0 或 1
## Build
...
...
@@ -23,8 +23,14 @@ CXX=nvcc make GPU_ARCH=sm_80
```
bash
# BW系列:
HIP_VISIBLE_DEVICES
=
1 numactl
-N
0
-m
0 ./gemv_bench
--verify
1
-M
11264
-K
4096
HIP_VISIBLE_DEVICES
=
1 numactl
-N
0
-m
0 ./gemv_bench
--verify
1
-M
11264
-K
4096
--alpha
1
--beta
0
-B
128
HIP_VISIBLE_DEVICES
=
1 numactl
-N
0
-m
0 ./gemv_bench
--verify
1
-M
4096
-K
11264
--alpha
1
--beta
0
-B
128
HIP_VISIBLE_DEVICES
=
1 numactl
-N
0
-m
0 ./gemv_bench
--verify
1
-M
12288
-K
4096
--alpha
1
--beta
0
-B
128
HIP_VISIBLE_DEVICES
=
1 numactl
-N
0
-m
0 ./gemv_bench
--verify
1
-M
4096
-K
4096
--alpha
1
--beta
1
-B
128
# A800:
./gemv_bench
--verify
1
-M
11264
-K
4096
./gemv_bench
--verify
1
-M
11264
-K
4096
--alpha
1
--beta
0
-B
256
./gemv_bench
--verify
1
-M
4096
-K
11264
--alpha
1
--beta
0
-B
256
./gemv_bench
--verify
1
-M
12288
-K
4096
--alpha
1
--beta
0
-B
256
./gemv_bench
--verify
1
-M
4096
-K
4096
--alpha
1
--beta
1
-B
256
```
\ No newline at end of file
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment