Skip to content

GitLab

Unverified Commit 43c898f6 authored by Chao Liu's avatar Chao Liu Committed by GitHub
Browse files

disable print for group conv multiple D (#421)

parent 370efa6c
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 3 deletions
include/ck/stream_config.hpp
View file @ 43c898f6
...@@ -10,4 +10,5 @@ struct StreamConfig ...@@ -10,4 +10,5 @@ struct StreamConfig
{ {
hipStream_t stream_id_ = nullptr; hipStream_t stream_id_ = nullptr;
bool time_kernel_ = false; bool time_kernel_ = false;
int log_level_ = 0;
}; };
include/ck/tensor_operation/gpu/device/device_grouped_conv_fwd_multiple_d_xdl_cshuffle.hpp
View file @ 43c898f6
...@@ -606,9 +606,11 @@ struct DeviceGroupedConvFwdMultipleD_Xdl_CShuffle ...@@ -606,9 +606,11 @@ struct DeviceGroupedConvFwdMultipleD_Xdl_CShuffle
float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{}) float Run(const Argument& arg, const StreamConfig& stream_config = StreamConfig{})
{ {
#if 1 if(stream_config.log_level_ > 0)
{
arg.Print(); arg.Print();
#endif }
if(!GridwiseGemm::CheckValidity(arg.a_grid_desc_m_k_, if(!GridwiseGemm::CheckValidity(arg.a_grid_desc_m_k_,
arg.b_grid_desc_n_k_, arg.b_grid_desc_n_k_,
arg.ds_grid_desc_m_n_, arg.ds_grid_desc_m_n_,
......
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