Use MakeQDramTileDistribution for q_dram_window

3e411ef0 · Qianfeng Zhang · ceea2722 · 3e411ef0
Commit 3e411ef0 authored Jan 31, 2025 by Qianfeng Zhang
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 .../ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp +6 -5

No files found.
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp
@@ -175,11 +175,12 @@ struct BlockFmhaPipelineQRKSVSAsync
        static_assert(NumKLdsBuffers >= 2);
        static_assert(NumVLdsBuffers >= 2);

-        auto q_dram_window = make_tile_window(q_dram_block_window_tmp.get_bottom_tensor_view(),
-                                              q_dram_block_window_tmp.get_window_lengths(),
-                                              q_dram_block_window_tmp.get_window_origin(),
-                                              Policy::template MakeQRegTileDistribution<Problem>());
-        auto original_q    = load_tile(q_dram_window);
+        auto q_dram_window =
+            make_tile_window(q_dram_block_window_tmp.get_bottom_tensor_view(),
+                             q_dram_block_window_tmp.get_window_lengths(),
+                             q_dram_block_window_tmp.get_window_origin(),
+                             Policy::template MakeQDramTileDistribution<Problem>());
+        auto original_q = load_tile(q_dram_window);

        __builtin_amdgcn_sched_barrier(0);