mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 13:11:25 +00:00
upgrade from clang-format-12 to clang-format-18 (#2568)
* upgrade to clang-format-18 * update to clang-format-18 in pre-commit-config
This commit is contained in:
@@ -385,7 +385,7 @@ struct FusedMoeGemmKernel
|
||||
auto o_window = [&]() {
|
||||
ODataType* o_ptr = reinterpret_cast<ODataType*>(kargs.o_ptr);
|
||||
auto o_view_ = make_naive_tensor_view<address_space_enum::global,
|
||||
memory_operation_enum::atomic_add>(
|
||||
memory_operation_enum::atomic_add>(
|
||||
o_ptr,
|
||||
make_tuple(kargs.num_tokens, kargs.hidden_size),
|
||||
make_tuple(kargs.stride_token, 1),
|
||||
|
||||
@@ -13,7 +13,7 @@
|
||||
namespace ck_tile {
|
||||
|
||||
#define MOE_SORTING_MOCK_ID(token_id_, topk_id_) \
|
||||
static_cast<uint32_t>(((token_id_)&0x00ffffff) | (((topk_id_)&0xff) << 24))
|
||||
static_cast<uint32_t>(((token_id_) & 0x00ffffff) | (((topk_id_) & 0xff) << 24))
|
||||
|
||||
#ifndef MOE_SORTING_USE_EX_KERNEL
|
||||
#define MOE_SORTING_USE_EX_KERNEL 1
|
||||
|
||||
@@ -267,8 +267,7 @@ struct FusedMoeGemmPipeline_FlatmmEx
|
||||
statically_indexed_array<a_thread_type, 2> as;
|
||||
|
||||
auto gld_a = [&]<typename PreNop = bool_constant<false>>(
|
||||
auto& a_store_, auto i_access, PreNop = {})
|
||||
{
|
||||
auto& a_store_, auto i_access, PreNop = {}) {
|
||||
async_load_tile_raw(a_store_, a_win, i_access, PreNop{});
|
||||
};
|
||||
auto move_a = [&]() {
|
||||
@@ -278,43 +277,40 @@ struct FusedMoeGemmPipeline_FlatmmEx
|
||||
load_tile_raw(a_, win_, i_access);
|
||||
};
|
||||
|
||||
auto gld_g = [&]<typename PreNop = bool_constant<false>>(
|
||||
auto& g_, auto i_access, PreNop = {})
|
||||
{
|
||||
if constexpr(IsGateOnly)
|
||||
{
|
||||
// TODO: hack!
|
||||
if constexpr(i_access.value == 0)
|
||||
auto gld_g =
|
||||
[&]<typename PreNop = bool_constant<false>>(auto& g_, auto i_access, PreNop = {}) {
|
||||
if constexpr(IsGateOnly)
|
||||
{
|
||||
g_win.bottom_tensor_view_ = g_view;
|
||||
// TODO: hack!
|
||||
if constexpr(i_access.value == 0)
|
||||
{
|
||||
g_win.bottom_tensor_view_ = g_view;
|
||||
}
|
||||
else if constexpr(i_access.value == issues_g / 2)
|
||||
{
|
||||
g_win.bottom_tensor_view_ = u_view;
|
||||
}
|
||||
}
|
||||
else if constexpr(i_access.value == issues_g / 2)
|
||||
{
|
||||
g_win.bottom_tensor_view_ = u_view;
|
||||
}
|
||||
}
|
||||
load_tile_raw(g_, g_win, i_access, FALSE, PreNop{});
|
||||
};
|
||||
load_tile_raw(g_, g_win, i_access, FALSE, PreNop{});
|
||||
};
|
||||
auto move_g = [&]() {
|
||||
move_tile_window(g_win, {number<0>{}, number<BlockShape::Block_Kr0>{}, number<0>{}});
|
||||
};
|
||||
statically_indexed_array<d_thread_type, 2> ds;
|
||||
|
||||
auto gld_d = [&]<typename PreNop = bool_constant<false>>(
|
||||
auto& d_, auto i_access, PreNop = {})
|
||||
{
|
||||
load_tile_raw(d_, d_win, i_access, FALSE, PreNop{});
|
||||
};
|
||||
auto gld_d =
|
||||
[&]<typename PreNop = bool_constant<false>>(auto& d_, auto i_access, PreNop = {}) {
|
||||
load_tile_raw(d_, d_win, i_access, FALSE, PreNop{});
|
||||
};
|
||||
auto move_d = [&]() {
|
||||
// d move along gemm-n
|
||||
move_tile_window(d_win, {number<BlockShape::Block_N1>{}, number<0>{}});
|
||||
};
|
||||
|
||||
auto atomic_add_o = [&]<typename PreNop = bool_constant<false>>(
|
||||
auto& o_, auto i_access, PreNop = {})
|
||||
{
|
||||
update_tile_raw(o_win, o_, i_access, TRUE, PreNop{});
|
||||
};
|
||||
auto atomic_add_o =
|
||||
[&]<typename PreNop = bool_constant<false>>(auto& o_, auto i_access, PreNop = {}) {
|
||||
update_tile_raw(o_win, o_, i_access, TRUE, PreNop{});
|
||||
};
|
||||
|
||||
auto acc_0 = Policy::template MakeCBlockTile_Gemm0<Problem>();
|
||||
auto acc_1s = generate_tuple(
|
||||
|
||||
Reference in New Issue
Block a user