From c1cfd4c894d46164f380429bded14918d47d28cf Mon Sep 17 00:00:00 2001 From: Chao Liu Date: Fri, 16 Sep 2022 09:46:32 -0500 Subject: [PATCH] disable print for group conv multiple D (#421) [ROCm/composable_kernel commit: 43c898f6ffe39244b6f023a565407a39ef2152bb] --- include/ck/stream_config.hpp | 1 + .../device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp | 8 +++++--- 2 files changed, 6 insertions(+), 3 deletions(-) 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_,