mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-15 11:34:54 +00:00
36 lines
858 B
Bash
Executable File
36 lines
858 B
Bash
Executable File
#!/bin/bash
|
|
EXE="$(find . -name tile_example_flatmm_basic -type f | head -n 1)"
|
|
KNAME=1
|
|
|
|
export CK_WARMUP=0
|
|
export CK_REPEAT=1
|
|
|
|
COMMON_ARGS='-v=2 '
|
|
|
|
run_tests() {
|
|
for m in 128 1024 8192; do
|
|
for n in 128 1024 8192; do
|
|
for k in 128 1024 8192; do
|
|
|
|
$EXE -m=$m -n=$n -k=$k -stride_a=0 -stride_b=0 -stride_c=0 -prec=$1 $COMMON_ARGS
|
|
if [ $? -eq 0 ]; then
|
|
echo "Success: Test with m=$m, n=$n, k=$k executed successfully."
|
|
else
|
|
echo "Error: Test with m=$m, n=$n, k=$k failed to execute properly."
|
|
# Optionally, exit or break if you need to halt further execution
|
|
# exit 1
|
|
fi
|
|
|
|
done
|
|
done
|
|
done
|
|
}
|
|
|
|
set -x
|
|
|
|
run_tests "bf16"
|
|
run_tests "fp16"
|
|
run_tests "fp8"
|
|
run_tests "bf8"
|
|
set +x
|