mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-02 04:31:25 +00:00
Minor Improvements in CK TILE memory copy EXAMPLE (#2678)
* Rename vector to ThreadTile * more notes on tile encoding * remove number<> from tuple of make_tile_window * add script to stress test the copy example
This commit is contained in:
50
example/ck_tile/39_copy/test_tile_example.sh
Executable file
50
example/ck_tile/39_copy/test_tile_example.sh
Executable file
@@ -0,0 +1,50 @@
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
|
||||
BIN="${BIN:-../../../build/bin/tile_example_copy}"
|
||||
WARMUP="${WARMUP:-20}"
|
||||
REPEAT="${REPEAT:-100}"
|
||||
VALIDATE="${VALIDATE:-1}"
|
||||
|
||||
MS=(128 256 512 1024)
|
||||
NS=(64 256 1024 2048 4096)
|
||||
PRECS=(fp16 fp32)
|
||||
|
||||
echo "Using BIN=$BIN"
|
||||
echo "WARMUP=$WARMUP REPEAT=$REPEAT VALIDATE=$VALIDATE"
|
||||
|
||||
failures=0
|
||||
|
||||
for prec in "${PRECS[@]}"; do
|
||||
for m in "${MS[@]}"; do
|
||||
for n in "${NS[@]}"; do
|
||||
echo "=============================================="
|
||||
echo "Running: prec=$prec m=$m n=$n"
|
||||
set +e
|
||||
out="$("$BIN" -prec="$prec" -m="$m" -n="$n" -warmup="$WARMUP" -repeat="$REPEAT" -v="$VALIDATE" 2>&1)"
|
||||
rc=$?
|
||||
set -e
|
||||
|
||||
echo "$out"
|
||||
if [[ $rc -ne 0 ]]; then
|
||||
echo "RUN ERROR (rc=$rc) for m=$m n=$n prec=$prec"
|
||||
((failures++)) || true
|
||||
continue
|
||||
fi
|
||||
|
||||
if [[ "$VALIDATE" == "1" ]]; then
|
||||
if ! grep -q "valid:y" <<<"$out"; then
|
||||
echo "VALIDATION FAILED for m=$m n=$n prec=$prec"
|
||||
((failures++)) || true
|
||||
fi
|
||||
fi
|
||||
done
|
||||
done
|
||||
done
|
||||
|
||||
echo "=============================================="
|
||||
if [[ $failures -eq 0 ]]; then
|
||||
echo "All runs passed"
|
||||
else
|
||||
echo "$failures runs failed"
|
||||
fi
|
||||
Reference in New Issue
Block a user