ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-18 17:48:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve profiler output.

Browse Source
This commit is contained in:
Ville Pietilä
2025-10-20 08:04:59 +00:00
parent dc65dc98e1
commit 3aec8b5493
2 changed files with 55 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
profiler/ck_tile/include/tile_profile_grouped_conv_bwd_weight_impl.hpp
View File

@@ -8,6 +8,7 @@
#include <iostream>
#include <iterator>
#include <typeinfo>
#include <sstream>
#include "ck_tile/host.hpp"
#include "ck_tile/host/convolution_parameter.hpp"
@@ -257,14 +258,33 @@ bool profile_grouped_conv_bwd_weight_impl(int do_verification,
}
}
std::cout << "\n********************************"
<< "\nBest configuration parameters:"
<< "\n********************************"
<< "\nname: " << best_op_name
<< "\navg_time: " << best_avg_time << "\ntflops: " << best_tflops
<< "\nGB/s: " << best_gb_per_sec
<< "\nSplitK: " << best_split_k
<< std::endl;
std::stringstream ss;
ss << "\n********************************"
<< "\nCK Tile best configuration parameters:"
<< "\n********************************"
<< "\nname: " << best_op_name
<< "\navg_time: " << best_avg_time << "\ntflops: " << best_tflops
<< "\nGB/s: " << best_gb_per_sec
<< "\nSplitK: " << best_split_k
<< std::endl;
std::cout << ss.str();
const char* log_file = std::getenv("CK_TILE_PROFILER_LOG_FILE");
if(log_file != nullptr)
{
std::ofstream out(log_file, std::ios::app);
if(out.is_open())
{
std::stringstream out_ss;
out_ss << "CK Tile best configuration:" << std::endl
<< "name: " << best_op_name << std::endl
<< "avg_time: " << best_avg_time << std::endl
<< "SplitK: " << best_split_k << std::endl;
out << out_ss.str();
out.close();
}
}
if(instance_index != -1)
{

30
profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp
View File

@@ -8,6 +8,7 @@
#include <iostream>
#include <iterator>
#include <typeinfo>
#include <sstream>
#include "ck/ck.hpp"
#include "ck/tensor_operation/gpu/device/tensor_layout.hpp"
@@ -330,9 +331,32 @@ bool profile_grouped_conv_bwd_weight_impl(int do_verification,
}
}
std::cout << "Best configuration parameters:" << "\nname: " << best_op_name
<< "\navg_time: " << best_avg_time << "\ntflops: " << best_tflops
<< "\nGB/s: " << best_gb_per_sec << ", SplitK " << best_split_k << std::endl;
std::stringstream ss;
ss << "\n********************************"
<< "\nCK best configuration parameters:"
<< "\n********************************"
<< "\nname: " << best_op_name
<< "\navg_time: " << best_avg_time << "\ntflops: " << best_tflops
<< "\nGB/s: " << best_gb_per_sec
<< "\nSplitK: " << best_split_k
<< std::endl;
std::cout << ss.str();
const char* log_file = std::getenv("CK_PROFILER_LOG_FILE");
if(log_file != nullptr)
{
std::ofstream out(log_file, std::ios::app);
if(out.is_open())
{
std::stringstream out_ss;
out_ss << "CK best configuration:" << std::endl
<< "name: " << best_op_name << std::endl
<< "avg_time: " << best_avg_time << std::endl
<< "SplitK: " << best_split_k << std::endl;
out << out_ss.str();
out.close();
}
}
if(instance_index != -1)
{