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
e7e73c8c
Commit
e7e73c8c
authored
Oct 10, 2023
by
Khalique Ahmed
Browse files
formatting
parent
b9d37172
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
5 additions
and
5 deletions
+5
-5
src/include/migraphx/op/allocate.hpp
src/include/migraphx/op/allocate.hpp
+1
-1
src/onnx/parse_constant_of_shape.cpp
src/onnx/parse_constant_of_shape.cpp
+1
-1
src/simplify_reshapes.cpp
src/simplify_reshapes.cpp
+2
-2
src/targets/gpu/device/include/migraphx/gpu/device/launch.hpp
...targets/gpu/device/include/migraphx/gpu/device/launch.hpp
+1
-1
No files found.
src/include/migraphx/op/allocate.hpp
View file @
e7e73c8c
...
@@ -65,7 +65,7 @@ struct allocate
...
@@ -65,7 +65,7 @@ struct allocate
{
{
migraphx
::
check_shapes
{
inputs
,
*
this
,
false
}.
has
(
1
).
only_dims
(
1
);
migraphx
::
check_shapes
{
inputs
,
*
this
,
false
}.
has
(
1
).
only_dims
(
1
);
const
auto
&
out_dims
=
inputs
.
at
(
0
);
const
auto
&
out_dims
=
inputs
.
at
(
0
);
std
::
size_t
max_val
=
std
::
numeric_limits
<
std
::
size_t
>::
max
();
std
::
size_t
max_val
=
std
::
numeric_limits
<
std
::
size_t
>::
max
();
std
::
vector
<
shape
::
dynamic_dimension
>
dyn_dims
(
out_dims
.
lens
().
at
(
0
),
std
::
vector
<
shape
::
dynamic_dimension
>
dyn_dims
(
out_dims
.
lens
().
at
(
0
),
shape
::
dynamic_dimension
{
0
,
max_val
});
shape
::
dynamic_dimension
{
0
,
max_val
});
return
{
buf_type
,
dyn_dims
};
return
{
buf_type
,
dyn_dims
};
...
...
src/onnx/parse_constant_of_shape.cpp
View file @
e7e73c8c
...
@@ -65,7 +65,7 @@ struct parse_constant_of_shape : op_parser<parse_constant_of_shape>
...
@@ -65,7 +65,7 @@ struct parse_constant_of_shape : op_parser<parse_constant_of_shape>
{
{
migraphx
::
shape
s
;
migraphx
::
shape
s
;
// input is empty, output is a scalar
// input is empty, output is a scalar
auto
type
=
l_val
.
get_shape
().
type
();
auto
type
=
l_val
.
get_shape
().
type
();
migraphx
::
argument
input
=
args
[
0
]
->
eval
();
migraphx
::
argument
input
=
args
[
0
]
->
eval
();
if
(
not
input
.
empty
())
if
(
not
input
.
empty
())
{
{
...
...
src/simplify_reshapes.cpp
View file @
e7e73c8c
...
@@ -647,8 +647,8 @@ struct find_broadcast_transpose
...
@@ -647,8 +647,8 @@ struct find_broadcast_transpose
{
{
auto
transpose
=
r
.
result
;
auto
transpose
=
r
.
result
;
auto
transpose_lens
=
transpose
->
get_shape
().
lens
();
auto
transpose_lens
=
transpose
->
get_shape
().
lens
();
auto
bcast_ins
=
r
.
instructions
[
"bcast_ins"
];
auto
bcast_ins
=
r
.
instructions
[
"bcast_ins"
];
auto
input
=
bcast_ins
->
inputs
().
front
();
auto
input
=
bcast_ins
->
inputs
().
front
();
// scalar transformation does not need extra transpose
// scalar transformation does not need extra transpose
if
(
not
input
->
get_shape
().
scalar
())
if
(
not
input
->
get_shape
().
scalar
())
{
{
...
...
src/targets/gpu/device/include/migraphx/gpu/device/launch.hpp
View file @
e7e73c8c
...
@@ -43,7 +43,7 @@ struct index
...
@@ -43,7 +43,7 @@ struct index
__device__
index_int
nglobal
()
const
{
return
blockDim
.
x
*
gridDim
.
x
;
}
// NOLINT
__device__
index_int
nglobal
()
const
{
return
blockDim
.
x
*
gridDim
.
x
;
}
// NOLINT
__device__
index_int
nlocal
()
const
{
return
blockDim
.
x
;
}
// NOLINT
__device__
index_int
nlocal
()
const
{
return
blockDim
.
x
;
}
// NOLINT
template
<
class
F
>
template
<
class
F
>
__device__
void
global_stride
(
index_int
n
,
F
f
)
const
__device__
void
global_stride
(
index_int
n
,
F
f
)
const
...
...
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