Skip to content

GitLab

"...composable_kernel_onnxruntime.git" did not exist on "4b616aad52807740908071e90e06e184d3177357"
Commit 1cce5d43 authored by Alan Turner's avatar Alan Turner
Browse files

Add export to tune script

parent 4e64e2c2
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tools/tune_ck.sh
View file @ 1cce5d43
...@@ -10,4 +10,5 @@ do ...@@ -10,4 +10,5 @@ do
MIGRAPHX_LOG_CK_GEMM=1 ./bin/driver run $MODEL -g --fill1 input_ids --input-dim @input_ids $N 384 | grep 'ck_gemm.*: \[{' | sort -u >> $LOG MIGRAPHX_LOG_CK_GEMM=1 ./bin/driver run $MODEL -g --fill1 input_ids --input-dim @input_ids $N 384 | grep 'ck_gemm.*: \[{' | sort -u >> $LOG
done done
python3 ../tools/tune_ck.py -n 16 -l $LOG -o $TUNING_DB python3 ../tools/tune_ck.py -n 16 -l $LOG -o $TUNING_DB
\ No newline at end of file export MIGRAPHX_CK_TUNING=$TUNING_DB
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment