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

fix some issues in async load

Browse Source
This commit is contained in:
joye
2025-05-19 08:56:03 +08:00
parent 41f5ccb0e6
commit f0e1dbca49
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async.hpp
View File

@@ -191,6 +191,8 @@ struct GemmPipelineAgBgCrCompAsync : public BaseGemmPipelineAgBgCrCompAsync<Prob
void* __restrict__ p_smem_0,
void* __restrict__ p_smem_1) const
{
ignore = a_element_func;
ignore = b_element_func;
static_assert(
std::is_same_v<ADataType, remove_cvref_t<typename ADramBlockWindowTmp::DataType>> &&
std::is_same_v<BDataType,

1
test/ck_tile/gemm/test_gemm_pipeline_kernel_types.hpp
View File

@@ -20,6 +20,7 @@ using Interwave = ck_tile::integral_constant<ck_tile::GemmPipelineScheduler,
using Mem = ck_tile::integral_constant<GemmPipelineType, GemmPipelineType::Mem>;
using CompV3 = ck_tile::integral_constant<GemmPipelineType, GemmPipelineType::CompV3>;
using CompV4 = ck_tile::integral_constant<GemmPipelineType, GemmPipelineType::CompV4>;
using Async = ck_tile::integral_constant<GemmPipelineType, GemmPipelineType::Async>;
// clang-format off
using KernelTypesMem = ::testing::Types<