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
MIGraphX
Commits
9c3073bd
Commit
9c3073bd
authored
Oct 17, 2022
by
Paul
Browse files
Format
parent
4fda0214
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
4 deletions
+3
-4
src/targets/gpu/jit/ck_gemm.cpp
src/targets/gpu/jit/ck_gemm.cpp
+3
-4
No files found.
src/targets/gpu/jit/ck_gemm.cpp
View file @
9c3073bd
...
@@ -144,8 +144,8 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
...
@@ -144,8 +144,8 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
auto
b_shape
=
inputs
[
1
];
auto
b_shape
=
inputs
[
1
];
auto
c_shape
=
inputs
[
2
];
auto
c_shape
=
inputs
[
2
];
auto
m
=
c_shape
.
lens
().
front
();
auto
m
=
c_shape
.
lens
().
front
();
auto
n
=
c_shape
.
lens
().
back
();
auto
n
=
c_shape
.
lens
().
back
();
auto
i
=
v
.
get
(
"tuning_val"
,
get_tuning_for
(
inputs
));
auto
i
=
v
.
get
(
"tuning_val"
,
get_tuning_for
(
inputs
));
const
auto
&
instance
=
get_instance
(
i
,
[
&
](
const
auto
&
x
)
->
bool
{
const
auto
&
instance
=
get_instance
(
i
,
[
&
](
const
auto
&
x
)
->
bool
{
...
@@ -163,8 +163,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
...
@@ -163,8 +163,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
options
.
kernel_name
=
"ck_gemm_kernel"
;
options
.
kernel_name
=
"ck_gemm_kernel"
;
options
.
virtual_inputs
=
inputs
;
options
.
virtual_inputs
=
inputs
;
auto
src
=
interpolate_string
(
ck_gemm_kernel
,
auto
src
=
interpolate_string
(
ck_gemm_kernel
,
{{
"instance"
,
join_strings
(
instance
,
","
)}});
{{
"instance"
,
join_strings
(
instance
,
","
)}});
return
compile_hip_code_object
(
src
,
options
);
return
compile_hip_code_object
(
src
,
options
);
}
}
...
...
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