squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Adding missing typename (#1191)

Browse Source 
Fixes clang build failures.
This commit is contained in:
Christian Sigg 2023-11-29 06:20:20 +01:00 committed by GitHub
parent eb01d5449d
commit 56fc3df03b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/41_fused_multi_head_attention/kernel_forward.h
View File

@ -733,7 +733,7 @@ struct AttentionKernel {
auto prologueV = [&](int blockN) { auto prologueV = [&](int blockN) {
typename MM1::Mma::IteratorB iterator_V( typename MM1::Mma::IteratorB iterator_V(
typename MM1::IteratorB::Params{MM1::LayoutB(p.v_strideM)}, typename MM1::IteratorB::Params{typename MM1::LayoutB(p.v_strideM)},
p.value_ptr + iter_key_start * p.v_strideM, p.value_ptr + iter_key_start * p.v_strideM,
{problem_size_1_k, problem_size_1_n}, {problem_size_1_k, problem_size_1_n},
thread_id(), thread_id(),
@ -997,7 +997,7 @@ struct AttentionKernel {
} }
typename MM1::Mma::IteratorB iterator_V( typename MM1::Mma::IteratorB iterator_V(
typename MM1::IteratorB::Params{MM1::LayoutB(p.v_strideM)}, typename MM1::IteratorB::Params{typename MM1::LayoutB(p.v_strideM)},
p.value_ptr + iter_key_start * p.v_strideM, p.value_ptr + iter_key_start * p.v_strideM,
{problem_size_1_k, problem_size_1_n}, {problem_size_1_k, problem_size_1_n},
thread_id(), thread_id(),