Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "b7abe77af693811fb1a17f1d95768487f5fbdb46"
Commit e86da0e9 authored by Po Yen Chen's avatar Po Yen Chen
Browse files

Fix wrong origin calculation

parent 60356c90
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_splitkv_pipeline_qr_ks_vs_async.hpp
View file @ e86da0e9
......@@ -445,7 +445,8 @@ struct BlockFmhaFwdSplitKVPipelineQRKSVSAsync
}
else if constexpr(BiasEnum == BlockAttentionBiasEnum::ALIBI)
{
const auto k_origin = k_dram_block_window.get_window_origin();
const auto k_origin = k_page_block_navigator.to_global_window_origin(
i_page_block_k, k_dram_block_window.get_window_origin());
constexpr auto s_spans = decltype(s_acc)::get_distributed_spans();
s_acc = tile_elementwise_in(s_acc_element_func, s_acc);
sweep_tile_span(s_spans[number<0>{}], [&](auto idx0) {
......
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