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
908ed025
Commit
908ed025
authored
Jun 03, 2019
by
Shucai Xiao
Browse files
clang format
parent
28ab5f76
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
6 deletions
+5
-6
src/targets/gpu/device/pack.cpp
src/targets/gpu/device/pack.cpp
+1
-1
src/targets/gpu/include/migraphx/gpu/quant_gemm.hpp
src/targets/gpu/include/migraphx/gpu/quant_gemm.hpp
+0
-1
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+4
-4
No files found.
src/targets/gpu/device/pack.cpp
View file @
908ed025
...
@@ -13,7 +13,7 @@ namespace device {
...
@@ -13,7 +13,7 @@ namespace device {
void
pack_a
(
hipStream_t
stream
,
const
argument
&
result
,
const
argument
&
arg
)
void
pack_a
(
hipStream_t
stream
,
const
argument
&
result
,
const
argument
&
arg
)
{
{
auto
comp_shape
=
arg
.
get_shape
();
auto
comp_shape
=
arg
.
get_shape
();
auto
out_lens
=
comp_shape
.
lens
();
auto
out_lens
=
comp_shape
.
lens
();
auto
dim_0
=
out_lens
.
size
()
-
2
;
auto
dim_0
=
out_lens
.
size
()
-
2
;
auto
dim_1
=
out_lens
.
size
()
-
1
;
auto
dim_1
=
out_lens
.
size
()
-
1
;
...
...
src/targets/gpu/include/migraphx/gpu/quant_gemm.hpp
View file @
908ed025
...
@@ -30,7 +30,6 @@ struct miopen_quant_gemm
...
@@ -30,7 +30,6 @@ struct miopen_quant_gemm
{
{
return
shapes
.
size
()
-
1
;
return
shapes
.
size
()
-
1
;
}
}
};
};
}
// namespace gpu
}
// namespace gpu
...
...
src/targets/gpu/lowering.cpp
View file @
908ed025
...
@@ -177,11 +177,11 @@ struct miopen_apply
...
@@ -177,11 +177,11 @@ struct miopen_apply
void
add_quant_dot_op
()
void
add_quant_dot_op
()
{
{
apply_map
.
emplace
(
"quant_dot"
,
[
=
](
instruction_ref
ins
)
{
apply_map
.
emplace
(
"quant_dot"
,
[
=
](
instruction_ref
ins
)
{
auto
&&
op
=
any_cast
<
op
::
quant_dot
>
(
ins
->
get_operator
());
auto
&&
op
=
any_cast
<
op
::
quant_dot
>
(
ins
->
get_operator
());
auto
inputs
=
ins
->
inputs
();
auto
inputs
=
ins
->
inputs
();
auto
in_shapes
=
to_shapes
(
inputs
);
auto
in_shapes
=
to_shapes
(
inputs
);
auto
arg_a
=
allocate_gpu
(
in_shapes
[
0
]);
auto
arg_a
=
allocate_gpu
(
in_shapes
[
0
]);
auto
arg_b
=
allocate_gpu
(
in_shapes
[
1
]);
auto
arg_b
=
allocate_gpu
(
in_shapes
[
1
]);
auto
quant_dot
=
miopen_quant_gemm
{
op
,
arg_a
,
arg_b
};
auto
quant_dot
=
miopen_quant_gemm
{
op
,
arg_a
,
arg_b
};
auto
output
=
insert_allocation
(
ins
,
ins
->
get_shape
());
auto
output
=
insert_allocation
(
ins
,
ins
->
get_shape
());
...
...
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