From 2074b3b8ab4c778d40284e0043760534e6dcc045 Mon Sep 17 00:00:00 2001 From: "lcy.seso" Date: Wed, 18 Sep 2024 05:16:10 +0000 Subject: [PATCH] Fixed ncu script bugs. --- artifacts/run_all_ncu_flash2.sh | 2 +- artifacts/run_all_ncu_triton.sh | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/artifacts/run_all_ncu_flash2.sh b/artifacts/run_all_ncu_flash2.sh index c2f8de1d6..913dcf599 100755 --- a/artifacts/run_all_ncu_flash2.sh +++ b/artifacts/run_all_ncu_flash2.sh @@ -9,5 +9,5 @@ mha_dir="$benchmark_dir/multi-head_attention/baseline" # 1. ncu test the mha benchmark echo "NCU profiling mha benchmark" $ncu_dir/ncu --section "MemoryWorkloadAnalysis" \ - --csv --set full python3 $stack_rnn_path/$mha_dir/test_pt_model.py > $log_dir/flash2_attention_ncu.csv + --csv --set full python3 $mha_dir/test_pt_model.py > $log_dir/flash2_attention_ncu.csv diff --git a/artifacts/run_all_ncu_triton.sh b/artifacts/run_all_ncu_triton.sh index adf355544..0498aa506 100755 --- a/artifacts/run_all_ncu_triton.sh +++ b/artifacts/run_all_ncu_triton.sh @@ -9,7 +9,7 @@ mha_dir="$benchmark_dir/multi-head_attention/baseline" # 1. ncu test the mha benchmark echo "NCU profiling mha benchmark" $ncu_dir/ncu --section "MemoryWorkloadAnalysis" \ - --csv --set full python3 $stack_rnn_path/$mha_dir/test_triton_model.py > $log_dir/triton_attention_ncu.csv + --csv --set full python3 $mha_dir/test_triton_model.py > $log_dir/triton_attention_ncu.csv bigbird_dir="$benchmark_dir/blocked_sparse_attention/triton" @@ -17,4 +17,4 @@ bigbird_dir="$benchmark_dir/blocked_sparse_attention/triton" echo "NCU profiling BigBird benchmark" $ncu_dir/ncu --section "MemoryWorkloadAnalysis" \ --csv --set full python3 $bigbird_dir/main.py \ - --default_test > $log_dir/triton_bigbird_ncu.csv \ No newline at end of file + --default_test True > $log_dir/triton_bigbird_ncu.csv \ No newline at end of file