Skip to content

GitLab

"composable_kernel/include/utility/sequence.hpp" did not exist on "8a4b59785b4f5ba48468d53618ca270c5da599a7"
Commit aa59ecaa authored by Max Podkorytov's avatar Max Podkorytov
Browse files

run clang-format==12

parent 82fb3f84
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 29 deletions
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp
View file @ aa59ecaa
...@@ -127,39 +127,25 @@ struct BlockFmhaPipelineQSKSVS ...@@ -127,39 +127,25 @@ struct BlockFmhaPipelineQSKSVS
typename OAccElementFunction, typename OAccElementFunction,
typename PositionEncoding> typename PositionEncoding>
CK_TILE_HOST_DEVICE auto CK_TILE_HOST_DEVICE auto
operator()(const QDramBlockWindowTmp & q_dram_block_window_tmp, // M0*K0 tile operator()(const QDramBlockWindowTmp& q_dram_block_window_tmp, // M0*K0 tile
const QElementFunction & const QElementFunction& q_element_func,
q_element_func, const KDramBlockWindowTmp& k_dram_block_window_tmp, // N0*K0 tile
const KDramBlockWindowTmp & const KElementFunction& k_element_func,
k_dram_block_window_tmp, // N0*K0 tile const VDramBlockWindowTmp& v_dram_block_window_tmp, // N1*K1 tile
const KElementFunction & const VElementFunction& v_element_func,
k_element_func, const BiasDramBlockWindowTmp& bias_dram_block_window_tmp, // M0*N0 tile
const VDramBlockWindowTmp & const BiasElementFunction& bias_element_func,
v_dram_block_window_tmp, // N1*K1 tile RandValDramBlockWindowTmp& randval_dram_block_window_tmp,
const VElementFunction & LSEDramBlockWindowTmp& lse_dram_window_tmp, // M0*1 tile
v_element_func, const LSEElementFunction& lse_element_func,
const BiasDramBlockWindowTmp & const SAccElementFunction& s_acc_element_func,
bias_dram_block_window_tmp, // M0*N0 tile const PComputeElementFunction& p_compute_element_func,
const BiasElementFunction & const OAccElementFunction& o_acc_element_func,
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, FmhaMask mask,
PositionEncoding position_encoding, PositionEncoding position_encoding,
float scale_s, float scale_s,
void* smem_ptr, void* smem_ptr,
DropoutType & DropoutType& dropout) const
dropout) const
{ {
static_assert( static_assert(
std::is_same_v<QDataType, remove_cvref_t<typename QDramBlockWindowTmp::DataType>> && std::is_same_v<QDataType, remove_cvref_t<typename QDramBlockWindowTmp::DataType>> &&
......
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