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
Commits
41ee82a3
Commit
41ee82a3
authored
Mar 23, 2023
by
aska-0096
Browse files
reproduce the scratch
parent
9d16e558
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
8 deletions
+9
-8
example/01_gemm/gemm_wmma_fp16.cpp
example/01_gemm/gemm_wmma_fp16.cpp
+7
-7
include/ck/tensor_operation/gpu/block/blockwise_gemm_wmma.hpp
...ude/ck/tensor_operation/gpu/block/blockwise_gemm_wmma.hpp
+2
-1
No files found.
example/01_gemm/gemm_wmma_fp16.cpp
View file @
41ee82a3
...
...
@@ -36,14 +36,14 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmWmma_CShuffle
CElementOp
,
GemmDefault
,
256
,
// BlockSize
256
,
// MPerBlock
32
,
// NPerBlock
128
,
// MPerBlock
128
,
// NPerBlock
32
,
// KPerBlock
8
,
// K1
16
,
// MPerWmma
16
,
// NPerWmma
4
,
// M Repeat
1
,
// N-Repeat
1
,
// M Repeat
8
,
// N-Repeat
S
<
4
,
64
,
1
>
,
S
<
1
,
0
,
2
>
,
S
<
1
,
0
,
2
>
,
...
...
@@ -51,15 +51,15 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmWmma_CShuffle
8
,
8
,
true
,
S
<
4
,
32
,
1
>
,
S
<
4
,
64
,
1
>
,
S
<
1
,
0
,
2
>
,
S
<
1
,
0
,
2
>
,
2
,
8
,
8
,
true
,
4
,
// C shuffle (M Repeat) Per store
1
,
// C shuffle (N Repeat) Per store
1
,
// C shuffle (M Repeat) Per store
4
,
// C shuffle (N Repeat) Per store
S
<
1
,
64
,
1
,
4
>
,
8
>
;
// clang-format on
...
...
include/ck/tensor_operation/gpu/block/blockwise_gemm_wmma.hpp
View file @
41ee82a3
...
...
@@ -396,7 +396,8 @@ struct BlockwiseGemmWMMA
A_K1
,
0x76543210
,
0xfedcba98
,
true
>
;
// true
false
>
;
};
template
<
bool
EnableLds
>
...
...
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