Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
liyinrong
composable_kernel
Commits
ba4edd2b
Commit
ba4edd2b
authored
3 years ago
by
Jing Zhang
Browse files
Options
Download
Email Patches
Plain Diff
fixed comment
parent
f24073a9
grouped_gemm_output_transpose
batched_gemm_output_transpose
grouped_gemm_bias_transpose
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
include/ck/tensor_operation/gpu/device/device_grouped_gemm_transpose_xdl.hpp
+6
-6
...peration/gpu/device/device_grouped_gemm_transpose_xdl.hpp
with
6 additions
and
6 deletions
+6
-6
include/ck/tensor_operation/gpu/device/device_grouped_gemm_transpose_xdl.hpp
View file @
ba4edd2b
...
...
@@ -54,9 +54,9 @@ __global__ void
}
GridwiseGemm
::
template
Run
<
HasMainKBlockLoop
>(
gemm_desc_ptr
[
group_id
].
a_ptr
,
gemm_desc_ptr
[
group_id
].
b_ptr
,
gemm_desc_ptr
[
group_id
].
c_ptr
,
gemm_desc_ptr
[
group_id
].
a_ptr
_
,
gemm_desc_ptr
[
group_id
].
b_ptr
_
,
gemm_desc_ptr
[
group_id
].
c_ptr
_
,
p_shared
,
gemm_desc_ptr
[
group_id
].
a_grid_desc_k0_m_k1_
,
gemm_desc_ptr
[
group_id
].
b_grid_desc_k0_n_k1_
,
...
...
@@ -347,9 +347,9 @@ struct DeviceGroupedGemmTransposeXdl : public DeviceGroupedGemmTranspose<AElemen
GroupedGemmBlock2CTileMap
grouped_gemm_transpose_block_2_ctile_map_
;
const
ADataType
*
a_ptr
;
const
BDataType
*
b_ptr
;
CDataType
*
c_ptr
;
const
ADataType
*
a_ptr
_
;
const
BDataType
*
b_ptr
_
;
CDataType
*
c_ptr
_
;
ck
::
index_t
BlockStart_
,
BlockEnd_
;
};
...
...
This diff is collapsed.
Click to expand it.
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
Menu
Projects
Groups
Snippets
Help