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
b91cb13a
Commit
b91cb13a
authored
Dec 11, 2024
by
Jing Zhang
Browse files
resolve merge conflicts
parents
bbb2cb69
57462a35
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
3 additions
and
4 deletions
+3
-4
example/01_gemm/gemm_xdl_bf16_pk_i4_v3.cpp
example/01_gemm/gemm_xdl_bf16_pk_i4_v3.cpp
+1
-1
example/01_gemm/gemm_xdl_fp16_pk_i4_v3.cpp
example/01_gemm/gemm_xdl_fp16_pk_i4_v3.cpp
+2
-3
No files found.
example/01_gemm/gemm_xdl_bf16_pk_i4_v3.cpp
View file @
b91cb13a
...
@@ -247,7 +247,7 @@ bool run_gemm_splitk_example(int argc, char* argv[])
...
@@ -247,7 +247,7 @@ bool run_gemm_splitk_example(int argc, char* argv[])
ProblemSizeSplitK
problem_size
;
ProblemSizeSplitK
problem_size
;
ExecutionConfig
config
;
ExecutionConfig
config
;
return
!
parse_cmd_args
(
argc
,
argv
,
problem_size
,
config
)
||
run_gemm
(
problem_size
,
config
);
return
parse_cmd_args
(
argc
,
argv
,
problem_size
,
config
)
&&
run_gemm
(
problem_size
,
config
);
}
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_splitk_example
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_splitk_example
(
argc
,
argv
);
}
example/01_gemm/gemm_xdl_fp16_pk_i4_v3.cpp
View file @
b91cb13a
// SPDX-License-Identifier: MIT
// SPDX-License-Identifier: MIT
// Copyright (c) 20
18-2023
, Advanced Micro Devices, Inc. All rights reserved.
// Copyright (c) 20
24
, Advanced Micro Devices, Inc. All rights reserved.
#include "common.hpp"
#include "common.hpp"
...
@@ -23,7 +23,6 @@ static constexpr auto GemmDefault = ck::tensor_operation::device::GemmSpecializa
...
@@ -23,7 +23,6 @@ static constexpr auto GemmDefault = ck::tensor_operation::device::GemmSpecializa
static
constexpr
bool
PermuteA
=
false
;
static
constexpr
bool
PermuteA
=
false
;
static
constexpr
bool
PermuteB
=
true
;
static
constexpr
bool
PermuteB
=
true
;
static
constexpr
ck
::
index_t
KPerBlock
=
128
;
static
constexpr
ck
::
index_t
KPerBlock
=
128
;
// clang-format off
// clang-format off
...
@@ -300,7 +299,7 @@ bool run_gemm_splitk_example(int argc, char* argv[])
...
@@ -300,7 +299,7 @@ bool run_gemm_splitk_example(int argc, char* argv[])
ProblemSizeSplitK
problem_size
;
ProblemSizeSplitK
problem_size
;
ExecutionConfig
config
;
ExecutionConfig
config
;
return
!
parse_cmd_args
(
argc
,
argv
,
problem_size
,
config
)
||
run_gemm
(
problem_size
,
config
);
return
parse_cmd_args
(
argc
,
argv
,
problem_size
,
config
)
&&
run_gemm
(
problem_size
,
config
);
}
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_splitk_example
(
argc
,
argv
);
}
int
main
(
int
argc
,
char
*
argv
[])
{
return
!
run_gemm_splitk_example
(
argc
,
argv
);
}
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