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

Remove more debug statements

Browse Source
This commit is contained in:
PoYen, Chen
2024-06-11 14:08:39 +00:00
parent 5efb80347e
commit 31505a2a04
1 changed files with 0 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
example/ck_tile/01_fmha/fmha_fwd.hpp
View File

@@ -142,12 +142,6 @@ struct fmha_fwd_args
std::tuple<uint64_t, uint64_t> drop_seed_offset;
};
#if defined(ENABLE_APP_DEBUG_STMTS)
#define APP_DEBUG_STMTS if(true)
#else
#define APP_DEBUG_STMTS if(false)
#endif
template <typename FmhaKernel>
auto fmha_fwd_create_kargs_and_grids(fmha_fwd_args args)
{
@@ -338,12 +332,6 @@ auto fmha_fwd_splitkv_create_kargs_and_grids(fmha_fwd_args args)
dim3 grids = FmhaFwdSplitKVKernel::GridSize(
args.batch, args.nhead_q, args.max_seqlen_q, args.hdim_v, args.num_splits);
APP_DEBUG_STMTS
{
std::cout << "splitkv grid size: " << grids.x << ", " << grids.y << ", " << grids.z
<< std::endl;
}
return ck_tile::make_tuple(kargs, grids);
}
@@ -394,12 +382,6 @@ auto fmha_fwd_splitkv_combine_create_kargs_and_grids(fmha_fwd_args args)
dim3 grids = FmhaFwdSplitKVCombineKernel::GridSize(
args.batch, args.nhead_q, args.max_seqlen_q, args.hdim_v);
APP_DEBUG_STMTS
{
std::cout << "splitkv-combine grid size: " << grids.x << ", " << grids.y << ", " << grids.z
<< std::endl;
}
return ck_tile::make_tuple(kargs, grids);
}