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
0328b06e
Commit
0328b06e
authored
Feb 13, 2025
by
Bartlomiej Kocot
Browse files
fixes
parent
7f179833
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
8 additions
and
0 deletions
+8
-0
example/ck_tile/03_gemm/universal_gemm.cpp
example/ck_tile/03_gemm/universal_gemm.cpp
+4
-0
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp
...tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp
+2
-0
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_mem.hpp
.../ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_mem.hpp
+2
-0
No files found.
example/ck_tile/03_gemm/universal_gemm.cpp
View file @
0328b06e
...
...
@@ -330,6 +330,7 @@ int run_gemm_example(int argc, char* argv[])
{
return
run_gemm_example_with_layouts
<
ck_tile
::
bf8_t
>
(
argc
,
argv
,
Row
{},
Col
{},
Row
{});
}
#if(CK_TILE_PIPELINE_DEFAULT == CK_TILE_PIPELINE_COMPUTE_V3)
else
if
(
data_type
==
"pk_int4_t"
)
{
// TODO: Add support for bhalf_t ADataType
...
...
@@ -337,6 +338,7 @@ int run_gemm_example(int argc, char* argv[])
ck_tile
::
pk_int4_t
,
ck_tile
::
half_t
>
(
argc
,
argv
,
Row
{},
Col
{},
Row
{});
}
#endif
else
{
throw
std
::
runtime_error
(
"Unsupported data_type!"
);
...
...
@@ -360,6 +362,7 @@ int run_gemm_example(int argc, char* argv[])
{
return
run_gemm_example_with_layouts
<
ck_tile
::
bf8_t
>
(
argc
,
argv
,
Col
{},
Col
{},
Row
{});
}
#if(CK_TILE_PIPELINE_DEFAULT == CK_TILE_PIPELINE_COMPUTE_V3)
else
if
(
data_type
==
"pk_int4_t"
)
{
// TODO: Add support for bhalf_t ADataType
...
...
@@ -367,6 +370,7 @@ int run_gemm_example(int argc, char* argv[])
ck_tile
::
pk_int4_t
,
ck_tile
::
half_t
>
(
argc
,
argv
,
Col
{},
Col
{},
Row
{});
}
#endif
else
{
throw
std
::
runtime_error
(
"Unsupported data_type!"
);
...
...
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_v4.hpp
View file @
0328b06e
...
...
@@ -60,6 +60,8 @@ struct GemmPipelineAgBgCrCompV4 : public BaseGemmPipelineAgBgCrCompV4<Problem>
using
CDataType
=
remove_cvref_t
<
typename
Problem
::
CDataType
>
;
using
BlockGemmShape
=
remove_cvref_t
<
typename
Problem
::
BlockGemmShape
>
;
static_assert
(
!
std
::
is_same_v
<
BDataType
,
pk_int4_t
>
,
"Not implemented"
);
static
constexpr
index_t
APackedSize
=
ck_tile
::
numeric_traits
<
remove_cvref_t
<
ADataType
>>::
PackedSize
;
static
constexpr
index_t
BPackedSize
=
...
...
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_mem.hpp
View file @
0328b06e
...
...
@@ -21,6 +21,8 @@ struct BaseGemmPipelineAgBgCrMem
using
BDataType
=
remove_cvref_t
<
typename
Problem
::
BDataType
>
;
using
BlockGemmShape
=
remove_cvref_t
<
typename
Problem
::
BlockGemmShape
>
;
static_assert
(
!
std
::
is_same_v
<
BDataType
,
pk_int4_t
>
,
"Not implemented"
);
static
constexpr
index_t
APackedSize
=
ck_tile
::
numeric_traits
<
remove_cvref_t
<
ADataType
>>::
PackedSize
;
static
constexpr
index_t
BPackedSize
=
...
...
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