mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-01 12:11:19 +00:00
ab5d027866
* 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>
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
|