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
9550f6e9
Commit
9550f6e9
authored
Jul 20, 2022
by
charlie
Browse files
Merge branch 'dyn_nms' of github.com:ROCmSoftwarePlatform/AMDMIGraphX into dyn_topK
parents
d7fb5892
0d8c92a0
Changes
34
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
18 additions
and
37 deletions
+18
-37
src/include/migraphx/onnx.hpp
src/include/migraphx/onnx.hpp
+10
-4
src/include/migraphx/op/allocate.hpp
src/include/migraphx/op/allocate.hpp
+1
-1
src/include/migraphx/op/batch_norm_inference.hpp
src/include/migraphx/op/batch_norm_inference.hpp
+0
-1
src/include/migraphx/op/convolution.hpp
src/include/migraphx/op/convolution.hpp
+0
-1
src/include/migraphx/op/gathernd.hpp
src/include/migraphx/op/gathernd.hpp
+1
-0
src/include/migraphx/op/get_tuple_elem.hpp
src/include/migraphx/op/get_tuple_elem.hpp
+1
-3
src/include/migraphx/op/identity.hpp
src/include/migraphx/op/identity.hpp
+1
-0
src/include/migraphx/op/if_op.hpp
src/include/migraphx/op/if_op.hpp
+0
-1
src/include/migraphx/op/leaky_relu.hpp
src/include/migraphx/op/leaky_relu.hpp
+0
-1
src/include/migraphx/op/loop.hpp
src/include/migraphx/op/loop.hpp
+0
-1
src/include/migraphx/op/nonmaxsuppression.hpp
src/include/migraphx/op/nonmaxsuppression.hpp
+1
-0
src/include/migraphx/op/nonzero.hpp
src/include/migraphx/op/nonzero.hpp
+1
-0
src/include/migraphx/op/prefix_scan_sum.hpp
src/include/migraphx/op/prefix_scan_sum.hpp
+0
-1
src/include/migraphx/op/quant_convolution.hpp
src/include/migraphx/op/quant_convolution.hpp
+0
-1
src/include/migraphx/op/quant_dot.hpp
src/include/migraphx/op/quant_dot.hpp
+0
-1
src/include/migraphx/op/reduce_op.hpp
src/include/migraphx/op/reduce_op.hpp
+1
-0
src/include/migraphx/op/reverse.hpp
src/include/migraphx/op/reverse.hpp
+0
-1
src/include/migraphx/op/scatter.hpp
src/include/migraphx/op/scatter.hpp
+1
-4
src/include/migraphx/op/scatter_add.hpp
src/include/migraphx/op/scatter_add.hpp
+0
-8
src/include/migraphx/op/scatter_mul.hpp
src/include/migraphx/op/scatter_mul.hpp
+0
-8
No files found.
src/include/migraphx/onnx.hpp
View file @
9550f6e9
...
@@ -33,13 +33,19 @@ inline namespace MIGRAPHX_INLINE_NS {
...
@@ -33,13 +33,19 @@ inline namespace MIGRAPHX_INLINE_NS {
/// struct to pass in onnx options to parser
/// struct to pass in onnx options to parser
struct
onnx_options
struct
onnx_options
{
{
/// Old way to set default fixed dimension size
(priority over default_dyn_dim_value)
/// Old way to set default fixed dimension size
std
::
size_t
default_dim_value
=
0
;
std
::
size_t
default_dim_value
=
0
;
/// Default dynamic dimension size (if not specified in onnx file)
/*!
* Default dynamic dimension size (if both default_dim_value and default_dyn_dim_value
* set parser throws)
*/
shape
::
dynamic_dimension
default_dyn_dim_value
=
{
1
,
1
,
0
};
shape
::
dynamic_dimension
default_dyn_dim_value
=
{
1
,
1
,
0
};
/// Explicitly specify the dims of an input
(priority over map_dyn_input_dims)
/// Explicitly specify the dims of an input
std
::
unordered_map
<
std
::
string
,
std
::
vector
<
std
::
size_t
>>
map_input_dims
=
{};
std
::
unordered_map
<
std
::
string
,
std
::
vector
<
std
::
size_t
>>
map_input_dims
=
{};
/// Explicitly specify dynamic dims of an input
/*!
* Explicitly specify dynamic dims of an input (if both map_input_dims and
* map_dyn_input_dims set parser throws)
*/
std
::
unordered_map
<
std
::
string
,
std
::
vector
<
shape
::
dynamic_dimension
>>
map_dyn_input_dims
=
{};
std
::
unordered_map
<
std
::
string
,
std
::
vector
<
shape
::
dynamic_dimension
>>
map_dyn_input_dims
=
{};
/// Continue parsing onnx file if an unknown operator is found
/// Continue parsing onnx file if an unknown operator is found
bool
skip_unknown_operators
=
false
;
bool
skip_unknown_operators
=
false
;
...
...
src/include/migraphx/op/allocate.hpp
View file @
9550f6e9
...
@@ -26,8 +26,8 @@
...
@@ -26,8 +26,8 @@
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/shape.hpp>
#include <migraphx/shape.hpp>
#include <migraphx/argument.hpp>
namespace
migraphx
{
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
src/include/migraphx/op/batch_norm_inference.hpp
View file @
9550f6e9
...
@@ -26,7 +26,6 @@
...
@@ -26,7 +26,6 @@
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/argument.hpp>
#include <cmath>
#include <cmath>
namespace
migraphx
{
namespace
migraphx
{
...
...
src/include/migraphx/op/convolution.hpp
View file @
9550f6e9
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/value.hpp>
#include <migraphx/value.hpp>
#include <migraphx/argument.hpp>
#include <cmath>
#include <cmath>
#include <utility>
#include <utility>
...
...
src/include/migraphx/op/gathernd.hpp
View file @
9550f6e9
...
@@ -27,6 +27,7 @@
...
@@ -27,6 +27,7 @@
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/argument.hpp>
namespace
migraphx
{
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
src/include/migraphx/op/get_tuple_elem.hpp
View file @
9550f6e9
...
@@ -26,10 +26,8 @@
...
@@ -26,10 +26,8 @@
#include "migraphx/errors.hpp"
#include "migraphx/errors.hpp"
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/stringutils.hpp>
#include <migraphx/streamutils.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/argument.hpp>
#include <utility>
#include <utility>
namespace
migraphx
{
namespace
migraphx
{
...
...
src/include/migraphx/op/identity.hpp
View file @
9550f6e9
...
@@ -25,6 +25,7 @@
...
@@ -25,6 +25,7 @@
#define MIGRAPHX_GUARD_OPERATORS_IDENTITY_HPP
#define MIGRAPHX_GUARD_OPERATORS_IDENTITY_HPP
#include <migraphx/op/unary.hpp>
#include <migraphx/op/unary.hpp>
#include <migraphx/argument.hpp>
namespace
migraphx
{
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
src/include/migraphx/op/if_op.hpp
View file @
9550f6e9
...
@@ -26,7 +26,6 @@
...
@@ -26,7 +26,6 @@
#include <array>
#include <array>
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/functional.hpp>
#include <migraphx/functional.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/module.hpp>
#include <migraphx/module.hpp>
...
...
src/include/migraphx/op/leaky_relu.hpp
View file @
9550f6e9
...
@@ -25,7 +25,6 @@
...
@@ -25,7 +25,6 @@
#define MIGRAPHX_GUARD_OPERATORS_LEAKY_RELU_HPP
#define MIGRAPHX_GUARD_OPERATORS_LEAKY_RELU_HPP
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
namespace
migraphx
{
namespace
migraphx
{
...
...
src/include/migraphx/op/loop.hpp
View file @
9550f6e9
...
@@ -25,7 +25,6 @@
...
@@ -25,7 +25,6 @@
#define MIGRAPHX_GUARD_OPERATORS_LOOP_HPP
#define MIGRAPHX_GUARD_OPERATORS_LOOP_HPP
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/functional.hpp>
#include <migraphx/functional.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/module.hpp>
#include <migraphx/module.hpp>
...
...
src/include/migraphx/op/nonmaxsuppression.hpp
View file @
9550f6e9
...
@@ -36,6 +36,7 @@
...
@@ -36,6 +36,7 @@
#include <migraphx/shape_for_each.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/output_iterator.hpp>
#include <migraphx/output_iterator.hpp>
#include <migraphx/argument.hpp>
namespace
migraphx
{
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
src/include/migraphx/op/nonzero.hpp
View file @
9550f6e9
...
@@ -29,6 +29,7 @@
...
@@ -29,6 +29,7 @@
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/float_equal.hpp>
#include <migraphx/float_equal.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/argument.hpp>
#include <cmath>
#include <cmath>
#include <utility>
#include <utility>
...
...
src/include/migraphx/op/prefix_scan_sum.hpp
View file @
9550f6e9
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <migraphx/operation.hpp>
#include <migraphx/operation.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/op/prefix_scan_op.hpp>
#include <migraphx/op/prefix_scan_op.hpp>
namespace
migraphx
{
namespace
migraphx
{
...
...
src/include/migraphx/op/quant_convolution.hpp
View file @
9550f6e9
...
@@ -28,7 +28,6 @@
...
@@ -28,7 +28,6 @@
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/value.hpp>
#include <migraphx/value.hpp>
#include <migraphx/argument.hpp>
#include <cmath>
#include <cmath>
#include <utility>
#include <utility>
...
...
src/include/migraphx/op/quant_dot.hpp
View file @
9550f6e9
...
@@ -25,7 +25,6 @@
...
@@ -25,7 +25,6 @@
#define MIGRAPHX_GUARD_OPERATORS_QUANT_DOT_HPP
#define MIGRAPHX_GUARD_OPERATORS_QUANT_DOT_HPP
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/gemm.hpp>
#include <migraphx/gemm.hpp>
#include <migraphx/value.hpp>
#include <migraphx/value.hpp>
...
...
src/include/migraphx/op/reduce_op.hpp
View file @
9550f6e9
...
@@ -27,6 +27,7 @@
...
@@ -27,6 +27,7 @@
#include <migraphx/op/name.hpp>
#include <migraphx/op/name.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/tensor_view.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/par_for.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
...
...
src/include/migraphx/op/reverse.hpp
View file @
9550f6e9
...
@@ -32,7 +32,6 @@
...
@@ -32,7 +32,6 @@
#include <migraphx/argument.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/argument.hpp>
#include <migraphx/value.hpp>
#include <migraphx/value.hpp>
namespace
migraphx
{
namespace
migraphx
{
...
...
src/include/migraphx/op/scatter.hpp
View file @
9550f6e9
...
@@ -26,15 +26,12 @@
...
@@ -26,15 +26,12 @@
#include <array>
#include <array>
#include <migraphx/check_shapes.hpp>
#include <migraphx/check_shapes.hpp>
#include <migraphx/stringutils.hpp>
#include <migraphx/streamutils.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/value.hpp>
#include <migraphx/value.hpp>
#include <migraphx/op/name.hpp>
#include <migraphx/op/name.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <cmath>
#include <migraphx/argument.hpp>
#include <utility>
namespace
migraphx
{
namespace
migraphx
{
inline
namespace
MIGRAPHX_INLINE_NS
{
inline
namespace
MIGRAPHX_INLINE_NS
{
...
...
src/include/migraphx/op/scatter_add.hpp
View file @
9550f6e9
...
@@ -25,15 +25,7 @@
...
@@ -25,15 +25,7 @@
#define MIGRAPHX_GUARD_OPERATORS_SCATTER_ADD_HPP
#define MIGRAPHX_GUARD_OPERATORS_SCATTER_ADD_HPP
#include <array>
#include <array>
#include <migraphx/check_shapes.hpp>
#include <migraphx/stringutils.hpp>
#include <migraphx/streamutils.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/value.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <cmath>
#include <utility>
#include <migraphx/op/scatter.hpp>
#include <migraphx/op/scatter.hpp>
// Scatter op. with "add" function as reduction.
// Scatter op. with "add" function as reduction.
...
...
src/include/migraphx/op/scatter_mul.hpp
View file @
9550f6e9
...
@@ -25,15 +25,7 @@
...
@@ -25,15 +25,7 @@
#define MIGRAPHX_GUARD_OPERATORS_SCATTER_MUL_HPP
#define MIGRAPHX_GUARD_OPERATORS_SCATTER_MUL_HPP
#include <array>
#include <array>
#include <migraphx/check_shapes.hpp>
#include <migraphx/stringutils.hpp>
#include <migraphx/streamutils.hpp>
#include <migraphx/shape_for_each.hpp>
#include <migraphx/config.hpp>
#include <migraphx/config.hpp>
#include <migraphx/value.hpp>
#include <migraphx/op/normalize_attribute.hpp>
#include <cmath>
#include <utility>
#include <migraphx/op/scatter.hpp>
#include <migraphx/op/scatter.hpp>
// Scatter op. with "multiply" as the reduction function.
// Scatter op. with "multiply" as the reduction function.
...
...
Prev
1
2
Next
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