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
a0ae1c61
Unverified
Commit
a0ae1c61
authored
May 09, 2024
by
Adam Osewski
Committed by
GitHub
May 09, 2024
Browse files
Fix MakeArgument (#1284)
parent
3c043cd1
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
3 deletions
+4
-3
example/15_grouped_gemm/grouped_gemm_multiple_d_xdl_fp16.cpp
example/15_grouped_gemm/grouped_gemm_multiple_d_xdl_fp16.cpp
+2
-1
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_xdl_cshuffle_tile_loop.hpp
...device_grouped_gemm_multiple_d_xdl_cshuffle_tile_loop.hpp
+2
-2
No files found.
example/15_grouped_gemm/grouped_gemm_multiple_d_xdl_fp16.cpp
View file @
a0ae1c61
...
@@ -92,9 +92,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co
...
@@ -92,9 +92,10 @@ bool run_grouped_gemm(const ProblemSize& problem_size, const ExecutionConfig& co
auto
group_count
=
problem_size
.
group_count
;
auto
group_count
=
problem_size
.
group_count
;
using
KernelArguments
=
ck
::
tensor_operation
::
device
::
GroupedGemmTileLoopKernelArguments
<
NumDs
>
;
using
KernelArguments
=
ck
::
tensor_operation
::
device
::
GroupedGemmTileLoopKernelArguments
<
NumDs
>
;
using
GemmDesc
=
ck
::
tensor_operation
::
device
::
GemmDesc
;
// GEMM shape
// GEMM shape
std
::
vector
<
ck
::
tensor_operation
::
device
::
GemmDesc
>
gemm_descs
;
std
::
vector
<
GemmDesc
>
gemm_descs
;
std
::
vector
<
KernelArguments
>
ggemm_kargs
;
std
::
vector
<
KernelArguments
>
ggemm_kargs
;
std
::
vector
<
void
*>
p_Cs
;
std
::
vector
<
void
*>
p_Cs
;
std
::
vector
<
const
void
*>
p_As
;
std
::
vector
<
const
void
*>
p_As
;
...
...
include/ck/tensor_operation/gpu/device/impl/device_grouped_gemm_multiple_d_xdl_cshuffle_tile_loop.hpp
View file @
a0ae1c61
...
@@ -375,7 +375,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
...
@@ -375,7 +375,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
std
::
vector
<
const
void
*>&
/* p_Bs */
,
std
::
vector
<
const
void
*>&
/* p_Bs */
,
std
::
vector
<
std
::
array
<
const
void
*
,
NumDTensor
>>&
/* p_Ds */
,
std
::
vector
<
std
::
array
<
const
void
*
,
NumDTensor
>>&
/* p_Ds */
,
std
::
vector
<
void
*>&
/* p_Es */
,
std
::
vector
<
void
*>&
/* p_Es */
,
std
::
vector
<
GemmDesc
>&
gemm_descs
,
const
std
::
vector
<
GemmDesc
>&
gemm_descs
,
AElementwiseOperation
a_element_op
,
AElementwiseOperation
a_element_op
,
BElementwiseOperation
b_element_op
,
BElementwiseOperation
b_element_op
,
CDEElementwiseOperation
cde_element_op
,
CDEElementwiseOperation
cde_element_op
,
...
@@ -643,7 +643,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
...
@@ -643,7 +643,7 @@ struct DeviceGroupedGemmMultipleDXdlCShuffleTileLoop
std
::
vector
<
const
void
*>&
p_Bs
,
std
::
vector
<
const
void
*>&
p_Bs
,
std
::
vector
<
std
::
array
<
const
void
*
,
NumDTensor
>>&
p_Ds
,
std
::
vector
<
std
::
array
<
const
void
*
,
NumDTensor
>>&
p_Ds
,
std
::
vector
<
void
*>&
p_Es
,
std
::
vector
<
void
*>&
p_Es
,
std
::
vector
<
GemmDesc
>
gemm_descs
,
std
::
vector
<
GemmDesc
>
&
gemm_descs
,
AElementwiseOperation
a_elementwise_op
,
AElementwiseOperation
a_elementwise_op
,
BElementwiseOperation
b_elementwise_op
,
BElementwiseOperation
b_elementwise_op
,
CDEElementwiseOperation
cde_elementwise_op
)
CDEElementwiseOperation
cde_elementwise_op
)
...
...
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