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
7bbb1cde
Commit
7bbb1cde
authored
Oct 08, 2022
by
Paul
Browse files
Format
parent
b57f58e1
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
3 deletions
+2
-3
src/targets/gpu/jit/ck_gemm.cpp
src/targets/gpu/jit/ck_gemm.cpp
+2
-3
No files found.
src/targets/gpu/jit/ck_gemm.cpp
View file @
7bbb1cde
...
@@ -39,7 +39,6 @@
...
@@ -39,7 +39,6 @@
#include <migraphx/pass_manager.hpp>
#include <migraphx/pass_manager.hpp>
#include <migraphx/env.hpp>
#include <migraphx/env.hpp>
const
std
::
vector
<
std
::
string
>&
const
std
::
vector
<
std
::
string
>&
get_instance
(
std
::
size_t
i
,
const
std
::
function
<
bool
(
const
std
::
vector
<
std
::
string
>&
)
>&
pred
);
get_instance
(
std
::
size_t
i
,
const
std
::
function
<
bool
(
const
std
::
vector
<
std
::
string
>&
)
>&
pred
);
...
@@ -100,7 +99,7 @@ static std::size_t get_grid_size(const std::vector<std::string>& s, std::size_t
...
@@ -100,7 +99,7 @@ static std::size_t get_grid_size(const std::vector<std::string>& s, std::size_t
return
int_div_ceil
(
m
,
mpb
)
*
int_div_ceil
(
n
,
npb
);
return
int_div_ceil
(
m
,
mpb
)
*
int_div_ceil
(
n
,
npb
);
}
}
template
<
class
F
,
class
Action
>
template
<
class
F
,
class
Action
>
auto
action_decorate
(
F
f
,
Action
action
)
auto
action_decorate
(
F
f
,
Action
action
)
{
{
return
[
=
](
auto
&&
...
xs
)
{
return
[
=
](
auto
&&
...
xs
)
{
...
@@ -169,7 +168,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
...
@@ -169,7 +168,7 @@ struct ck_gemm_compiler : compiler<ck_gemm_compiler>
{
{
auto
shapes
=
to_shapes
(
ins
->
inputs
());
auto
shapes
=
to_shapes
(
ins
->
inputs
());
return
action_decorate
(
replace
(
compile_op
(
ctx
,
shapes
,
op
.
to_value
())),
[
=
]
{
return
action_decorate
(
replace
(
compile_op
(
ctx
,
shapes
,
op
.
to_value
())),
[
=
]
{
if
(
enabled
(
MIGRAPHX_LOG_CK_GEMM
{}))
if
(
enabled
(
MIGRAPHX_LOG_CK_GEMM
{}))
std
::
cout
<<
"ck_gemm: "
<<
to_json_string
(
to_value
(
shapes
))
<<
std
::
endl
;
std
::
cout
<<
"ck_gemm: "
<<
to_json_string
(
to_value
(
shapes
))
<<
std
::
endl
;
});
});
}
}
...
...
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