ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

run clang-format==12

Browse Source
This commit is contained in:
Max Podkorytov
2024-12-19 20:10:38 +00:00
parent 82fb3f84fb
commit aa59ecaa22
1 changed files with 15 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp
View File

@@ -127,39 +127,25 @@ struct BlockFmhaPipelineQSKSVS
typename OAccElementFunction,
typename PositionEncoding>
CK_TILE_HOST_DEVICE auto
operator()(const QDramBlockWindowTmp & q_dram_block_window_tmp, // M0*K0 tile
const QElementFunction &
q_element_func,
const KDramBlockWindowTmp &
k_dram_block_window_tmp, // N0*K0 tile
const KElementFunction &
k_element_func,
const VDramBlockWindowTmp &
v_dram_block_window_tmp, // N1*K1 tile
const VElementFunction &
v_element_func,
const BiasDramBlockWindowTmp &
bias_dram_block_window_tmp, // M0*N0 tile
const BiasElementFunction &
bias_element_func,
RandValDramBlockWindowTmp &
randval_dram_block_window_tmp,
LSEDramBlockWindowTmp &
lse_dram_window_tmp, // M0*1 tile
const LSEElementFunction &
lse_element_func,
const SAccElementFunction &
s_acc_element_func,
const PComputeElementFunction &
p_compute_element_func,
const OAccElementFunction &
o_acc_element_func,
operator()(const QDramBlockWindowTmp& q_dram_block_window_tmp, // M0*K0 tile
const QElementFunction& q_element_func,
const KDramBlockWindowTmp& k_dram_block_window_tmp, // N0*K0 tile
const KElementFunction& k_element_func,
const VDramBlockWindowTmp& v_dram_block_window_tmp, // N1*K1 tile
const VElementFunction& v_element_func,
const BiasDramBlockWindowTmp& bias_dram_block_window_tmp, // M0*N0 tile
const BiasElementFunction& bias_element_func,
RandValDramBlockWindowTmp& randval_dram_block_window_tmp,
LSEDramBlockWindowTmp& lse_dram_window_tmp, // M0*1 tile
const LSEElementFunction& lse_element_func,
const SAccElementFunction& s_acc_element_func,
const PComputeElementFunction& p_compute_element_func,
const OAccElementFunction& o_acc_element_func,
FmhaMask mask,
PositionEncoding position_encoding,
float scale_s,
void* smem_ptr,
DropoutType &
dropout) const
DropoutType& dropout) const
{
static_assert(
std::is_same_v<QDataType, remove_cvref_t<typename QDramBlockWindowTmp::DataType>> &&