Skip to content

GitLab

Commit 6ebaa81e authored by Adam Osewski's avatar Adam Osewski
Browse files

Fix usage of CK_ENV

parent 129e58ae
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 15 deletions
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_splitk_xdl_cshuffle_tile_loop.hpp
View file @ 6ebaa81e
......@@ -787,7 +787,7 @@ struct DeviceGroupedGemmMultipleDSplitKXdlCShuffle
if((ck::type_convert<ck::index_t>(arg.gemm_kernel_args_.size()) +
arg.skipped_group_count_) != arg.group_count_)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "The group count is not equal to sum of skipped groups "
"and kernel args size!"
......@@ -811,7 +811,7 @@ struct DeviceGroupedGemmMultipleDSplitKXdlCShuffle
arg.K_BATCH);
if(not group_arg_valid)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "[" << __func__ << "] group id: " << i
<< " has invalid GridwiseGemm settings!" << std::endl;
......
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_xdl_splitk_cshuffle_v2.hpp
View file @ 6ebaa81e
......@@ -415,7 +415,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(!(M % MPerBlock == 0))
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg M value is not a multiple of MPerBlock! M: " << M << " "
<< __FILE__ << ":" << __LINE__ << ", in function: " << __func__
......@@ -429,7 +429,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(!(N % NPerBlock == 0))
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg N value is not a multiple of NPerBlock! N: " << N << " "
<< __FILE__ << ":" << __LINE__ << ", in function: " << __func__
......@@ -444,7 +444,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(!(K % KPerBlock == 0))
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg K value is not a multiple of ! KPerBlock: " << K << " "
<< __FILE__ << ":" << __LINE__ << ", in function: " << __func__
......@@ -458,7 +458,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(K % ABlockTransferSrcScalarPerVector != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg K (" << K
<< ") value is not a multiple of ABlockTransferSrcScalarPerVector ("
......@@ -472,7 +472,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(M % ABlockTransferSrcScalarPerVector != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg M (" << M
<< ") value is not a multiple of ABlockTransferSrcScalarPerVector ("
......@@ -487,7 +487,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(N % BBlockTransferSrcScalarPerVector != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg N (" << N
<< ") value is not a multiple of BBlockTransferSrcScalarPerVector ("
......@@ -501,7 +501,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(K % BBlockTransferSrcScalarPerVector != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg K (" << K
<< ") value is not a multiple of BBlockTransferSrcScalarPerVector ("
......@@ -516,7 +516,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(N % CDEShuffleBlockTransferScalarPerVector_NPerBlock != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg N (" << N
<< ") value is not a multiple of "
......@@ -532,7 +532,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
{
if(M % CDEShuffleBlockTransferScalarPerVector_NPerBlock != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "Arg M (" << M
<< ") value is not a multiple of "
......@@ -550,7 +550,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
if(k_batch_size < KPerBlock)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "The k-batch size (" << k_batch_size
<< ") value is less than KPerBlock!\n"
......@@ -562,7 +562,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
if(k_batch_size % KPerBlock != 0)
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "The k-batch size (" << k_batch_size
<< ") value is not a multiple of KPerBlock!\n"
......@@ -582,7 +582,7 @@ class GridwiseGemmMultipleD_xdl_splitk_cshuffle_v2
if(!GridwiseGemmPipe::IsSupported(num_k_loop))
{
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "The number of k loops (" << num_k_loop
<< ") value is not supported by GridwiseGemm Pipeline."
......
profiler/include/profiler/profile_grouped_gemm_multiple_d_splitk_impl.hpp
View file @ 6ebaa81e
......@@ -84,7 +84,7 @@ bool profile_ggemm_multid_splitk(int do_verification,
c_m_n_host_results.push_back(
Tensor<CDataType>(f_host_tensor_descriptor(Ms[i], Ns[i], StrideCs[i], CLayout{})));
if(ck::EnvIsEnabled(ENV(CK_LOGGING)))
if(ck::EnvIsEnabled(CK_ENV(CK_LOGGING)))
{
std::cout << "group: " << i << " a_m_k[" << i << "]:" << a_m_k[i].mDesc << ", b_k_n["
<< i << "]:" << b_k_n[i].mDesc << ", c_m_n_device_results[" << i
......
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