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

fixed vector load siz for fp4

Browse Source
This commit is contained in:
Sami Remes
2026-01-16 12:04:34 -05:00
parent 16ca5cb532
commit f09e10936d
7 changed files with 66 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/ops/gemm/kernel/universal_gemm_kernel.hpp
View File

@@ -673,8 +673,8 @@ struct UniversalGemmKernel
using AiLayout = remove_cvref_t<std::tuple_element_t<i.value, AsLayout>>;
using AiDataType = remove_cvref_t<std::tuple_element_t<i.value, AsDataType>>;
static_assert(GemmPipeline::GetVectorSizeA() == GemmPipeline::GetVectorSizeB(), "Vector size of A and B must be the same!");
static_assert(GemmPipeline::GetVectorSizeA() == 16, "Vector size of A must be 16!");
static_assert(GemmPipeline::GetVectorSizeB() == 16, "Vector size of B must be 16!");
static_assert(GemmPipeline::GetVectorSizeA() == 32, "Vector size of A must be 16!");
static_assert(GemmPipeline::GetVectorSizeB() == 32, "Vector size of B must be 16!");
if constexpr(std::is_same_v<AiLayout, tensor_layout::gemm::RowMajor>)
{
return make_naive_tensor_view<address_space_enum::global>(