diff --git a/include/ck/stream_config.hpp b/include/ck/stream_config.hpp index 95076606c4..70ca34555a 100644 --- a/include/ck/stream_config.hpp +++ b/include/ck/stream_config.hpp @@ -10,4 +10,5 @@ struct StreamConfig { hipStream_t stream_id_ = nullptr; bool time_kernel_ = false; + int log_level_ = 0; }; diff --git a/include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp b/include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp index 2e22aee225..03c17e6e76 100644 --- a/include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp +++ b/include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp @@ -606,9 +606,11 @@ struct DeviceGroupedConvFwdMultipleD_Xdl_CShuffle float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{}) { -#if 1 - arg.Print(); -#endif + if(stream_config.log_level_ > 0) + { + arg.Print(); + } + if(!GridwiseGemm::CheckValidity(arg.a_grid_desc_m_k_, arg.b_grid_desc_n_k_, arg.ds_grid_desc_m_n_,