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
1f12c4e0
Commit
1f12c4e0
authored
Dec 03, 2024
by
Po Yen Chen
Browse files
Update comment
parent
3361b802
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp
include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp
+2
-2
No files found.
include/ck_tile/ops/fmha/kernel/fmha_fwd_splitkv_kernel.hpp
View file @
1f12c4e0
...
...
@@ -639,7 +639,7 @@ struct FmhaFwdSplitKVKernel
if
constexpr
(
kIsPagedKV
)
{
constexpr
index_t
vector_size
=
FmhaPipeline
::
kAlignmentK
;
// (hdim_q/vector_size,
seqlen_k
, vector_size)
// (hdim_q/vector_size,
page_block_size
, vector_size)
const
auto
view
=
make_naive_tensor_view
<
address_space_enum
::
global
>
(
data
,
// will update this pointer if using paged-kvcache
make_tuple
(
kargs
.
hdim_q
/
vector_size
,
height
,
number
<
vector_size
>
{}),
...
...
@@ -647,7 +647,7 @@ struct FmhaFwdSplitKVKernel
number
<
vector_size
>
{},
number
<
1
>
{});
// (
seqlen_k
, hdim_q)
// (
page_block_size
, hdim_q)
return
transform_tensor_view
(
view
,
make_tuple
(
make_pass_through_transform
(
height
),
...
...
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