mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-14 18:17:44 +00:00
0aee5c2d16
* Support bf16/fb8/bf8 datatypes for ck_tile/gemm
* remove commented out code.
* Addressing code review comments and enabling universal_gemm for all the supported data types.
* Merge conflict resolution.
* Solve the memory pipeline compilation error. Merge with the new change of CShuffle
* finish the feature, pass the tests
* Fix the pipeline and add the benchmark script for other data types
---------
Co-authored-by: ThomasNing <thomas.ning@amd.com>
[ROCm/composable_kernel commit: ab5d027866]
15 lines
411 B
Bash
Executable File
15 lines
411 B
Bash
Executable File
#!/bin/sh
|
|
EXE="$(find . -name tile_example_gemm_basic -type f | head -n 1)"
|
|
VALID=1
|
|
|
|
|
|
for b_matrix_layout in "C"; do
|
|
for m in "64" "512" "1024" "2048"; do
|
|
for n in "512" "1024" "2048"; do
|
|
for k in "64" "512" "1024" "2048"; do
|
|
$EXE -prec=fp16 -m=$m -n=$n -k=$k -a_layout="R" -b_layout="$b_matrix_layout" -c_layout="R" -v=$VALID
|
|
done
|
|
done
|
|
done
|
|
done
|