Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
gaoqiong
composable_kernel_ROCM
Commits
73a4d827
Commit
73a4d827
authored
Dec 29, 2024
by
Po Yen Chen
Browse files
Only update tensor view attributes if change page-block
parent
3d167b4b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
include/ck_tile/ops/fmha/block/page_block_navigator.hpp
include/ck_tile/ops/fmha/block/page_block_navigator.hpp
+9
-5
No files found.
include/ck_tile/ops/fmha/block/page_block_navigator.hpp
View file @
73a4d827
...
@@ -144,11 +144,15 @@ struct PageBlockNavigator
...
@@ -144,11 +144,15 @@ struct PageBlockNavigator
const
WindowOrigin
local_window_origin
=
to_local_window_origin
(
global_window_origin
);
const
WindowOrigin
local_window_origin
=
to_local_window_origin
(
global_window_origin
);
const
index_t
new_block_index
=
get_block_index
(
global_window_origin
);
const
index_t
new_block_index
=
get_block_index
(
global_window_origin
);
/// TODO: only update necessary attributes
if
(
block_index
!=
new_block_index
)
tile_window
.
bottom_tensor_view_
.
desc_
=
{
(
is_last_block
(
new_block_index
)
?
last_view
:
complete_view
).
get_tensor_descriptor
();
/// TODO: only update necessary attributes
tile_window
.
set_window_origin
(
local_window_origin
);
tile_window
.
bottom_tensor_view_
.
desc_
=
tile_window
.
set_bottom_tensor_view_data_ptr
(
get_block_ptr
(
new_block_index
));
(
is_last_block
(
new_block_index
)
?
last_view
:
complete_view
)
.
get_tensor_descriptor
();
tile_window
.
set_window_origin
(
local_window_origin
);
tile_window
.
set_bottom_tensor_view_data_ptr
(
get_block_ptr
(
new_block_index
));
}
return
new_block_index
;
return
new_block_index
;
}
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment