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
cf86db72
Commit
cf86db72
authored
Nov 07, 2018
by
Paul
Browse files
Merge branch 'master' into fp16
parents
af454aeb
414e2fac
Changes
167
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
97 additions
and
0 deletions
+97
-0
src/include/migraph/config.hpp
src/include/migraph/config.hpp
+12
-0
src/include/migraph/constant_propagate.hpp
src/include/migraph/constant_propagate.hpp
+3
-0
src/include/migraph/context.hpp
src/include/migraph/context.hpp
+3
-0
src/include/migraph/dead_code_elimination.hpp
src/include/migraph/dead_code_elimination.hpp
+3
-0
src/include/migraph/dfor.hpp
src/include/migraph/dfor.hpp
+4
-0
src/include/migraph/eliminate_allocation.hpp
src/include/migraph/eliminate_allocation.hpp
+5
-0
src/include/migraph/eliminate_concat.hpp
src/include/migraph/eliminate_concat.hpp
+24
-0
src/include/migraph/eliminate_contiguous.hpp
src/include/migraph/eliminate_contiguous.hpp
+3
-0
src/include/migraph/env.hpp
src/include/migraph/env.hpp
+3
-0
src/include/migraph/erase.hpp
src/include/migraph/erase.hpp
+3
-0
src/include/migraph/errors.hpp
src/include/migraph/errors.hpp
+3
-0
src/include/migraph/fallthrough.hpp
src/include/migraph/fallthrough.hpp
+4
-0
src/include/migraph/float_equal.hpp
src/include/migraph/float_equal.hpp
+3
-0
src/include/migraph/functional.hpp
src/include/migraph/functional.hpp
+3
-0
src/include/migraph/fwd_conv_batchnorm_rewrite.hpp
src/include/migraph/fwd_conv_batchnorm_rewrite.hpp
+3
-0
src/include/migraph/generate.hpp
src/include/migraph/generate.hpp
+3
-0
src/include/migraph/half.hpp
src/include/migraph/half.hpp
+3
-0
src/include/migraph/instruction.hpp
src/include/migraph/instruction.hpp
+6
-0
src/include/migraph/instruction_ref.hpp
src/include/migraph/instruction_ref.hpp
+3
-0
src/include/migraph/iterator_for.hpp
src/include/migraph/iterator_for.hpp
+3
-0
No files found.
src/include/migraph/config.hpp
0 → 100644
View file @
cf86db72
#ifndef MIGRAPH_GUARD_CONFIG_HPP
#define MIGRAPH_GUARD_CONFIG_HPP
namespace
migraph
{
#ifndef MIGRAPH_USE_CLANG_TIDY
#define MIGRAPH_INLINE_NS version_1
#endif
}
// namespace migraph
#endif
src/include/migraph/constant_propagate.hpp
View file @
cf86db72
...
...
@@ -2,8 +2,10 @@
#define MIGRAPH_GUARD_RTGLIB_CONSTANT_PROPAGATE_HPP
#include <string>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
...
...
@@ -13,6 +15,7 @@ struct constant_propagate
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/context.hpp
View file @
cf86db72
...
...
@@ -7,8 +7,10 @@
#include <memory>
#include <type_traits>
#include <utility>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
#ifdef DOXYGEN
...
...
@@ -203,6 +205,7 @@ inline const ValueType& any_cast(const context& x)
#endif
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/dead_code_elimination.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,10 @@
#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
...
...
@@ -14,6 +16,7 @@ struct dead_code_elimination
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/dfor.hpp
View file @
cf86db72
#ifndef MIGRAPH_GUARD_MIGRAPHLIB_DFOR_HPP
#define MIGRAPH_GUARD_MIGRAPHLIB_DFOR_HPP
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
// Multidimensional for loop
inline
auto
dfor
()
...
...
@@ -20,6 +23,7 @@ auto dfor(T x, Ts... xs)
};
}
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/eliminate_allocation.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,11 @@
#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
struct
eliminate_allocation
...
...
@@ -14,6 +17,8 @@ struct eliminate_allocation
std
::
string
name
()
const
{
return
"eliminate_allocation"
;
}
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/eliminate_concat.hpp
0 → 100644
View file @
cf86db72
#ifndef MIGRAPH_GUARD_RTGLIB_ELIMINATE_CONCAT_HPP
#define MIGRAPH_GUARD_RTGLIB_ELIMINATE_CONCAT_HPP
#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/concat_opt.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
struct
eliminate_concat
{
concat_optimization
concat_opt
;
std
::
string
name
()
const
{
return
"eliminate_concat"
;
}
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/eliminate_contiguous.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,10 @@
#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
...
...
@@ -14,6 +16,7 @@ struct eliminate_contiguous
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/env.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,10 @@
#include <vector>
#include <string>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
// Declare a cached environment variable
#define MIGRAPH_DECLARE_ENV_VAR(x) \
...
...
@@ -31,6 +33,7 @@ bool disabled(T)
return
result
;
}
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/erase.hpp
View file @
cf86db72
...
...
@@ -2,8 +2,10 @@
#define MIGRAPH_GUARD_ERASE_HPP
#include <algorithm>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
/**
* @brief Erase all elements from a container
...
...
@@ -31,6 +33,7 @@ auto erase_if(R&& r, P&& pred)
return
r
.
erase
(
std
::
remove_if
(
r
.
begin
(),
r
.
end
(),
pred
),
r
.
end
());
}
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/errors.hpp
View file @
cf86db72
...
...
@@ -4,8 +4,10 @@
#include <exception>
#include <stdexcept>
#include <string>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
/// Represents exceptions that can be thrown by migraphlib
struct
exception
:
std
::
runtime_error
...
...
@@ -44,6 +46,7 @@ inline std::string make_source_context(const std::string& file, int line)
#define MIGRAPH_THROW(...) \
throw migraph::make_exception(migraph::make_source_context(__FILE__, __LINE__), __VA_ARGS__)
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/fallthrough.hpp
View file @
cf86db72
#ifndef MIGRAPH_GUARD_FALLTHROUGH_HPP
#define MIGRAPH_GUARD_FALLTHROUGH_HPP
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
#ifdef __clang__
#define MIGRAPH_FALLTHROUGH [[clang::fallthrough]]
...
...
@@ -9,6 +12,7 @@ namespace migraph {
#define MIGRAPH_FALLTHROUGH
#endif
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/float_equal.hpp
View file @
cf86db72
...
...
@@ -9,8 +9,10 @@
#endif
#include <migraph/requires.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
template
<
class
...
Ts
>
using
common_type
=
typename
std
::
common_type
<
Ts
...
>::
type
;
...
...
@@ -40,6 +42,7 @@ struct float_equal_fn
static
constexpr
float_equal_fn
float_equal
{};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/functional.hpp
View file @
cf86db72
...
...
@@ -2,8 +2,10 @@
#define MIGRAPH_GUARD_RTGLIB_FUNCTIONAL_HPP
#include <utility>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
swallow
{
...
...
@@ -129,6 +131,7 @@ auto fold(F f)
return
[
=
](
auto
&&
...
xs
)
{
return
fold_impl
(
f
,
std
::
forward
<
decltype
(
xs
)
>
(
xs
)...);
};
}
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/fwd_conv_batchnorm_rewrite.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,10 @@
#include <string>
#include <migraph/instruction_ref.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
program
;
...
...
@@ -14,6 +16,7 @@ struct fwd_conv_batchnorm_rewrite
void
apply
(
program
&
p
)
const
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/generate.hpp
View file @
cf86db72
...
...
@@ -4,9 +4,11 @@
#include <migraph/argument.hpp>
#include <migraph/literal.hpp>
#include <migraph/type_traits.hpp>
#include <migraph/config.hpp>
#include <random>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
template
<
class
T
,
MIGRAPH_REQUIRES
(
is_floating_point
<
T
>{})
>
constexpr
T
normalize
(
unsigned
long
z
)
...
...
@@ -91,6 +93,7 @@ literal generate_literal(shape s, unsigned long seed = 0);
literal
abs
(
literal
l
);
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/half.hpp
View file @
cf86db72
...
...
@@ -9,8 +9,10 @@
#define MIGRAPH_GUARD_RTGLIB_HALF_HPP
#include <half.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
using
half
=
half_float
::
half
;
...
...
@@ -31,6 +33,7 @@ struct deduce<half_float::detail::expr>
template
<
class
T
>
using
deduce
=
typename
detail
::
deduce
<
T
>::
type
;
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/instruction.hpp
View file @
cf86db72
...
...
@@ -6,10 +6,12 @@
#include <migraph/instruction_ref.hpp>
#include <migraph/operation.hpp>
#include <migraph/erase.hpp>
#include <migraph/config.hpp>
#include <string>
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
shape
compute_shape
(
const
operation
&
op
,
const
std
::
vector
<
instruction_ref
>&
args
);
...
...
@@ -69,6 +71,8 @@ struct instruction
static
void
replace
(
instruction_ref
ins
,
operation
o
,
const
shape
&
r
,
std
::
vector
<
instruction_ref
>
args
);
static
instruction_ref
get_output_alias
(
instruction_ref
ins
);
private:
// internal
void
replace
(
operation
o
,
const
shape
&
r
,
std
::
vector
<
instruction_ref
>
args
);
...
...
@@ -86,6 +90,7 @@ struct instruction
std
::
vector
<
instruction_ref
>
arguments
;
literal
lit
;
};
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
namespace
std
{
...
...
@@ -99,6 +104,7 @@ struct hash<migraph::instruction_ref>
return
std
::
hash
<
migraph
::
instruction
*>
{}(
&*
x
);
}
};
}
// namespace std
#endif
src/include/migraph/instruction_ref.hpp
View file @
cf86db72
...
...
@@ -3,12 +3,15 @@
#include <list>
#include <functional>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
struct
instruction
;
using
instruction_ref
=
std
::
list
<
instruction
>::
iterator
;
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/include/migraph/iterator_for.hpp
View file @
cf86db72
...
...
@@ -3,8 +3,10 @@
#include <cassert>
#include <type_traits>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
template
<
class
T
>
struct
iterator_for_range
...
...
@@ -37,6 +39,7 @@ iterator_for_range<T> iterator_for(T& x)
return
{
&
x
};
}
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
Prev
1
2
3
4
5
6
…
9
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