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
b8e153a4
Commit
b8e153a4
authored
Mar 27, 2023
by
aska-0096
Browse files
a fix
parent
44be6438
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
5 additions
and
5 deletions
+5
-5
example/02_gemm_bilinear/gemm_bilinear_wmma_fp16.cpp
example/02_gemm_bilinear/gemm_bilinear_wmma_fp16.cpp
+3
-3
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
...ation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
+2
-2
No files found.
example/02_gemm_bilinear/gemm_bilinear_wmma_fp16.cpp
View file @
b8e153a4
...
@@ -87,8 +87,8 @@ using DeviceOpInstance =
...
@@ -87,8 +87,8 @@ using DeviceOpInstance =
8
,
8
,
16
,
16
,
16
,
16
,
8
,
4
,
1
,
2
,
S
<
4
,
64
,
1
>
,
S
<
4
,
64
,
1
>
,
S
<
1
,
0
,
2
>
,
S
<
1
,
0
,
2
>
,
S
<
1
,
0
,
2
>
,
S
<
1
,
0
,
2
>
,
...
@@ -105,7 +105,7 @@ using DeviceOpInstance =
...
@@ -105,7 +105,7 @@ using DeviceOpInstance =
true
,
true
,
1
,
1
,
1
,
1
,
S
<
1
,
16
,
1
,
16
>
,
S
<
1
,
32
,
1
,
8
>
,
8
>
;
8
>
;
int
main
(
int
argc
,
char
*
argv
[])
int
main
(
int
argc
,
char
*
argv
[])
...
...
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
View file @
b8e153a4
...
@@ -151,7 +151,7 @@ __global__ void
...
@@ -151,7 +151,7 @@ __global__ void
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx1100__) || defined(__gfx1101__) || \
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx1100__) || defined(__gfx1101__) || \
defined(__gfx1102__))
defined(__gfx1102__))
// printf("entry kernel launch");
// printf("entry kernel launch");
__shared__
char
p_shared
[
GridwiseOp
::
Get
SharedMem
oryNumberOfByte
()
];
__shared__
char
p_shared
[
GridwiseOp
::
SharedMem
Trait
::
lds_size
];
const
index_t
num_blocks_per_batch
=
const
index_t
num_blocks_per_batch
=
__builtin_amdgcn_readfirstlane
(
get_grid_size
()
/
batch_count
);
__builtin_amdgcn_readfirstlane
(
get_grid_size
()
/
batch_count
);
...
@@ -241,7 +241,7 @@ __global__ void
...
@@ -241,7 +241,7 @@ __global__ void
{
{
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx1100__) || defined(__gfx1101__) || \
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx1100__) || defined(__gfx1101__) || \
defined(__gfx1102__))
defined(__gfx1102__))
__shared__
char
p_shared
[
GridwiseOp
::
Get
SharedMem
oryNumberOfByte
()
];
__shared__
char
p_shared
[
GridwiseOp
::
SharedMem
Trait
::
lds_size
];
GridwiseOp
::
template
Run
<
HasMainKBlockLoop
>(
p_a_grid
,
GridwiseOp
::
template
Run
<
HasMainKBlockLoop
>(
p_a_grid
,
p_b_grid
,
p_b_grid
,
...
...
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