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
a88810da
Commit
a88810da
authored
Oct 11, 2022
by
charlie
Browse files
Fix things
convolution revert
parent
d9d2215a
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
15 additions
and
1 deletion
+15
-1
src/common.cpp
src/common.cpp
+2
-0
src/include/migraphx/op/convolution.hpp
src/include/migraphx/op/convolution.hpp
+5
-1
src/shape.cpp
src/shape.cpp
+8
-0
No files found.
src/common.cpp
View file @
a88810da
...
...
@@ -135,6 +135,7 @@ instruction_ref insert_common_op(module& m,
a_input
=
m
.
insert_instruction
(
ins
,
make_op
(
"convert"
,
{{
"target_type"
,
c_type
}}),
a_input
);
}
return
a_input
;
});
}
else
...
...
@@ -151,6 +152,7 @@ instruction_ref insert_common_op(module& m,
input
=
m
.
insert_instruction
(
ins
,
make_op
(
"convert"
,
{{
"target_type"
,
common
.
type
()}}),
input
);
}
return
input
;
});
}
return
m
.
insert_instruction
(
ins
,
op
,
inputs
);
...
...
src/include/migraphx/op/convolution.hpp
View file @
a88810da
...
...
@@ -176,7 +176,11 @@ struct convolution
auto
min_spatial_dims
=
calc_conv_lens
(
x_shape
.
min_lens
(),
w_shape
.
max_lens
());
auto
max_spatial_dims
=
calc_conv_lens
(
x_shape
.
max_lens
(),
w_shape
.
min_lens
());
auto
opt_spatial_dims
=
calc_conv_lens
(
x_shape
.
opt_lens
(),
w_shape
.
opt_lens
());
return
shape
{
x_shape
.
type
(),
min_spatial_dims
,
max_spatial_dims
,
opt_spatial_dims
};
for
(
size_t
i
=
0
;
i
<
num_spatial_dims
;
++
i
)
{
output_dyn_dims
.
push_back
(
shape
::
dynamic_dimension
{
min_spatial_dims
[
i
],
max_spatial_dims
[
i
],
opt_spatial_dims
[
i
]});
}
}
return
shape
{
x_shape
.
type
(),
output_dyn_dims
};
}
...
...
src/shape.cpp
View file @
a88810da
...
...
@@ -237,6 +237,14 @@ shape::shape(type_t t, std::vector<shape::dynamic_dimension> dims)
{
}
shape
::
shape
(
type_t
t
,
std
::
vector
<
std
::
size_t
>
mins
,
std
::
vector
<
std
::
size_t
>
maxes
,
std
::
vector
<
std
::
size_t
>
opts
)
:
impl
(
std
::
make_shared
<
shape_impl
>
(
t
,
std
::
move
(
mins
),
std
::
move
(
maxes
),
std
::
move
(
opts
)))
{
}
shape
::
shape
(
const
std
::
vector
<
shape
>&
subs
)
:
impl
(
std
::
make_shared
<
shape_impl
>
(
subs
))
{}
shape
::
shape
(
std
::
shared_ptr
<
shape_impl
>
pimpl
)
:
impl
(
std
::
move
(
pimpl
))
{}
...
...
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