ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 19:57:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix file formatting

Browse Source
This commit is contained in:
Aleksander Dudek
2025-10-02 05:28:18 -05:00
parent 87f8319753
commit 89cd7b482b
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck_tile/core/arch/generic_memory_space_atomic.hpp
View File

@@ -350,7 +350,7 @@ CK_TILE_DEVICE void atomic_add<fp16x2_t>(fp16x2_t* p_dst, fp16x2_t const& x)
new_.f162 = add_f16x2_t(cur_v.f162, x);
new_v = new_.u32;
cur_v.u32 = atomicCAS(dword_addr.u32_a, old_v, new_v);
} while(cur_v.u32 != old_v);
} while(cur_v.u32 != old_v);
#endif
}

6
include/ck_tile/ops/epilogue/cshuffle_epilogue.hpp
View File

@@ -169,8 +169,10 @@ struct CShuffleEpilogue
using DiDataType = remove_cvref_t<std::tuple_element_t<index.value, DsDataType>>;
using DiLayout = remove_cvref_t<std::tuple_element_t<index.value, DsLayout>>;
static_assert(std::is_same_v<DiLayout, ELayout>, "ELayout is not equal to DiLayout!");
constexpr bool IsDRowMajor = std::is_same_v<DiLayout, tensor_layout::gemm::RowMajor> ? true : false;
constexpr bool IsDColMajor = std::is_same_v<DiLayout, tensor_layout::gemm::ColumnMajor> ? true : false;
constexpr bool IsDRowMajor =
std::is_same_v<DiLayout, tensor_layout::gemm::RowMajor> ? true : false;
constexpr bool IsDColMajor =
std::is_same_v<DiLayout, tensor_layout::gemm::ColumnMajor> ? true : false;
if constexpr(IsDRowMajor)
{
return std::min(static_cast<int>(NPerIteration),