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
1b895f4a
Commit
1b895f4a
authored
Jul 19, 2019
by
Shucai Xiao
Browse files
clang format
parent
0d877311
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
11 additions
and
11 deletions
+11
-11
src/targets/gpu/gemm.cpp
src/targets/gpu/gemm.cpp
+10
-10
src/tf/tf.cpp
src/tf/tf.cpp
+1
-1
No files found.
src/targets/gpu/gemm.cpp
View file @
1b895f4a
...
@@ -174,12 +174,12 @@ shape miopen_gemm::compute_shape(const std::vector<shape>& inputs) const
...
@@ -174,12 +174,12 @@ shape miopen_gemm::compute_shape(const std::vector<shape>& inputs) const
auto
a_strides
=
inputs
[
0
].
strides
();
auto
a_strides
=
inputs
[
0
].
strides
();
if
(
a_strides
.
size
()
>
2
)
if
(
a_strides
.
size
()
>
2
)
{
{
auto
dim_1
=
a_strides
.
size
()
-
1
;
auto
dim_1
=
a_strides
.
size
()
-
1
;
auto
dim_0
=
dim_1
-
1
;
auto
dim_0
=
dim_1
-
1
;
auto
matrix_size
=
std
::
max
(
a_strides
[
dim_0
],
a_strides
[
1
]);
auto
matrix_size
=
std
::
max
(
a_strides
[
dim_0
],
a_strides
[
1
]);
if
(
std
::
adjacent_find
(
a_strides
.
begin
(),
a_strides
.
begin
()
+
dim_0
,
[
&
](
auto
i
,
auto
j
)
{
if
(
std
::
adjacent_find
(
a_strides
.
begin
(),
a_strides
.
begin
()
+
dim_0
,
[
&
](
auto
i
,
auto
j
)
{
return
(
i
<
j
or
i
<
matrix_size
or
j
<
matrix_size
);
return
(
i
<
j
or
i
<
matrix_size
or
j
<
matrix_size
);
})
!=
a_strides
.
begin
()
+
dim_0
)
})
!=
a_strides
.
begin
()
+
dim_0
)
{
{
MIGRAPHX_THROW
(
"DOT: batch size of a {"
+
to_string_range
(
a_strides
)
+
MIGRAPHX_THROW
(
"DOT: batch size of a {"
+
to_string_range
(
a_strides
)
+
"} is transposed!"
);
"} is transposed!"
);
...
@@ -189,12 +189,12 @@ shape miopen_gemm::compute_shape(const std::vector<shape>& inputs) const
...
@@ -189,12 +189,12 @@ shape miopen_gemm::compute_shape(const std::vector<shape>& inputs) const
auto
b_strides
=
inputs
[
1
].
strides
();
auto
b_strides
=
inputs
[
1
].
strides
();
if
(
b_strides
.
size
()
>
2
)
if
(
b_strides
.
size
()
>
2
)
{
{
auto
dim_1
=
b_strides
.
size
()
-
1
;
auto
dim_1
=
b_strides
.
size
()
-
1
;
auto
dim_0
=
dim_1
-
1
;
auto
dim_0
=
dim_1
-
1
;
auto
matrix_size
=
std
::
max
(
b_strides
[
dim_0
],
b_strides
[
1
]);
auto
matrix_size
=
std
::
max
(
b_strides
[
dim_0
],
b_strides
[
1
]);
if
(
std
::
adjacent_find
(
b_strides
.
begin
(),
b_strides
.
begin
()
+
dim_0
,
[
&
](
auto
i
,
auto
j
)
{
if
(
std
::
adjacent_find
(
b_strides
.
begin
(),
b_strides
.
begin
()
+
dim_0
,
[
&
](
auto
i
,
auto
j
)
{
return
(
i
<
j
or
i
<
matrix_size
or
j
<
matrix_size
);
return
(
i
<
j
or
i
<
matrix_size
or
j
<
matrix_size
);
})
!=
b_strides
.
begin
()
+
dim_0
)
})
!=
b_strides
.
begin
()
+
dim_0
)
{
{
MIGRAPHX_THROW
(
"DOT: batch size of b {"
+
to_string_range
(
b_strides
)
+
MIGRAPHX_THROW
(
"DOT: batch size of b {"
+
to_string_range
(
b_strides
)
+
"} is transposed!"
);
"} is transposed!"
);
...
...
src/tf/tf.cpp
View file @
1b895f4a
...
@@ -721,7 +721,7 @@ struct tf_parser
...
@@ -721,7 +721,7 @@ struct tf_parser
const
attribute_map
&
attributes
,
const
attribute_map
&
attributes
,
std
::
vector
<
instruction_ref
>
args
)
std
::
vector
<
instruction_ref
>
args
)
{
{
int
axis
=
-
1
;
int
axis
=
-
1
;
auto
num_dims
=
args
[
0
]
->
get_shape
().
lens
().
size
();
auto
num_dims
=
args
[
0
]
->
get_shape
().
lens
().
size
();
if
(
contains
(
attributes
,
"axis"
))
if
(
contains
(
attributes
,
"axis"
))
{
{
...
...
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