Skip to content

GitLab

"git@developer.sourcefind.cn:modelzoo/resnet50_tensorflow.git" did not exist on "a684cbb8543d2c25ddc7341420d68f8642124e3d"
Commit 3e411ef0 authored by Qianfeng Zhang's avatar Qianfeng Zhang
Browse files

Use MakeQDramTileDistribution for q_dram_window

parent ceea2722
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 5 deletions
include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
View file @ 3e411ef0
...@@ -175,11 +175,12 @@ struct BlockFmhaPipelineQRKSVSAsync ...@@ -175,11 +175,12 @@ struct BlockFmhaPipelineQRKSVSAsync
static_assert(NumKLdsBuffers >= 2); static_assert(NumKLdsBuffers >= 2);
static_assert(NumVLdsBuffers >= 2); static_assert(NumVLdsBuffers >= 2);
auto q_dram_window = make_tile_window(q_dram_block_window_tmp.get_bottom_tensor_view(), auto q_dram_window =
q_dram_block_window_tmp.get_window_lengths(), make_tile_window(q_dram_block_window_tmp.get_bottom_tensor_view(),
q_dram_block_window_tmp.get_window_origin(), q_dram_block_window_tmp.get_window_lengths(),
Policy::template MakeQRegTileDistribution<Problem>()); q_dram_block_window_tmp.get_window_origin(),
auto original_q = load_tile(q_dram_window); Policy::template MakeQDramTileDistribution<Problem>());
auto original_q = load_tile(q_dram_window);
__builtin_amdgcn_sched_barrier(0); __builtin_amdgcn_sched_barrier(0);
......
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