Commit 1507fcfb authored by Aviral Goel's avatar Aviral Goel
Browse files

added stash and unstash for perf logs

parent 3e75a4dd
...@@ -356,7 +356,7 @@ def cmake_build(Map conf=[:]){ ...@@ -356,7 +356,7 @@ def cmake_build(Map conf=[:]){
stash includes: "perf_tile_flex_attn_**_fp16_gfx90a.log", name: "perf_tile_flex_attn_log_gfx90a" stash includes: "perf_tile_flex_attn_**_fp16_gfx90a.log", name: "perf_tile_flex_attn_log_gfx90a"
} }
else if (arch_type == 2){ else if (arch_type == 2){
stash includes: "perf_tile_flex_attn_**_fp16_gfx942.log", name: "perf_tile_flex_attn__gfx942" stash includes: "perf_tile_flex_attn_**_fp16_gfx942.log", name: "perf_tile_flex_attn_log_gfx942"
} }
} }
catch(Exception err){ catch(Exception err){
...@@ -677,6 +677,15 @@ def process_results(Map conf=[:]){ ...@@ -677,6 +677,15 @@ def process_results(Map conf=[:]){
echo "could not locate the FMHA performance logs: ${err.getMessage()}." echo "could not locate the FMHA performance logs: ${err.getMessage()}."
} }
} }
if (params.RUN_CK_TILE_FLEX_ATTENTION_TESTS){
try{
unstash "perf_tile_flex_attn_log_gfx90a"
unstash "perf_tile_flex_attn_log_gfx942"
}
catch(Exception err){
echo "could not locate the FMHA performance logs: ${err.getMessage()}."
}
}
if (params.RUN_CK_TILE_GEMM_TESTS){ if (params.RUN_CK_TILE_GEMM_TESTS){
try{ try{
unstash "perf_tile_gemm_log_gfx942" unstash "perf_tile_gemm_log_gfx942"
......
...@@ -35,7 +35,7 @@ function print_log_header(){ ...@@ -35,7 +35,7 @@ function print_log_header(){
example/ck_tile/18_flexattn/script/smoke_test_fwd.sh example/ck_tile/18_flexattn/script/smoke_test_fwd.sh
#run performance benchmarks #run performance benchmarks
export fmha_fwd_log="perf_fmha_fwd_$GPU_arch.log" export fmha_fwd_log="perf_tile_flex_attn_$GPU_arch.log"
print_log_header $fmha_fwd_log $env_type $branch $host_name print_log_header $fmha_fwd_log $env_type $branch $host_name
example/ck_tile/18_flexattn/script/benchmark_fwd.sh 2>&1 | tee -a $fmha_fwd_log example/ck_tile/18_flexattn/script/benchmark_fwd.sh 2>&1 | tee -a $fmha_fwd_log
......
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