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
4c031df7
".github/workflows/release-docker-amd-nightly.yml" did not exist on "4075677621f3be941f205cac669d37b8db3a8851"
Commit
4c031df7
authored
Nov 09, 2018
by
wsttiger
Browse files
Fixed conflicts
parents
d32653a5
ed5f9897
Changes
164
Show whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
58 additions
and
40 deletions
+58
-40
src/targets/gpu/include/migraph/gpu/lowering.hpp
src/targets/gpu/include/migraph/gpu/lowering.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/miopen.hpp
src/targets/gpu/include/migraph/gpu/miopen.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/mul.hpp
src/targets/gpu/include/migraph/gpu/mul.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/pooling.hpp
src/targets/gpu/include/migraph/gpu/pooling.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/relu.hpp
src/targets/gpu/include/migraph/gpu/relu.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/rocblas.hpp
src/targets/gpu/include/migraph/gpu/rocblas.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/softmax.hpp
src/targets/gpu/include/migraph/gpu/softmax.hpp
+3
-1
src/targets/gpu/include/migraph/gpu/target.hpp
src/targets/gpu/include/migraph/gpu/target.hpp
+4
-0
src/targets/gpu/include/migraph/gpu/write_literals.hpp
src/targets/gpu/include/migraph/gpu/write_literals.hpp
+2
-1
src/targets/gpu/leaky_relu.cpp
src/targets/gpu/leaky_relu.cpp
+2
-1
src/targets/gpu/lowering.cpp
src/targets/gpu/lowering.cpp
+2
-0
src/targets/gpu/mul.cpp
src/targets/gpu/mul.cpp
+2
-1
src/targets/gpu/pooling.cpp
src/targets/gpu/pooling.cpp
+2
-1
src/targets/gpu/relu.cpp
src/targets/gpu/relu.cpp
+2
-1
src/targets/gpu/rocblas.cpp
src/targets/gpu/rocblas.cpp
+2
-1
src/targets/gpu/softmax.cpp
src/targets/gpu/softmax.cpp
+2
-1
src/targets/gpu/target.cpp
src/targets/gpu/target.cpp
+2
-0
src/targets/gpu/write_literals.cpp
src/targets/gpu/write_literals.cpp
+3
-1
test/auto_contiguous_test.cpp
test/auto_contiguous_test.cpp
+7
-14
test/common_subexpression_elimination_test.cpp
test/common_subexpression_elimination_test.cpp
+5
-11
No files found.
src/targets/gpu/include/migraph/gpu/lowering.hpp
View file @
4c031df7
...
...
@@ -2,9 +2,11 @@
#define MIGRAPH_GUARD_RTGLIB_MIOPEN_LOWERING_HPP
#include <migraph/program.hpp>
#include <migraph/config.hpp>
#include <migraph/gpu/context.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
lowering
...
...
@@ -15,7 +17,7 @@ struct lowering
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/miopen.hpp
View file @
4c031df7
...
...
@@ -4,8 +4,10 @@
#include <migraph/manage_ptr.hpp>
#include <migraph/operators.hpp>
#include <miopen/miopen.h>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
using
miopen_handle
=
MIGRAPH_MANAGE_PTR
(
miopenHandle_t
,
miopenDestroy
);
...
...
@@ -115,7 +117,7 @@ inline fused_operator_args make_fused_args()
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/mul.hpp
View file @
4c031df7
...
...
@@ -7,6 +7,7 @@
#include <migraph/operators.hpp>
#include <migraph/generate.hpp>
#include <migraph/shape_for_each.hpp>
#include <migraph/config.hpp>
#include <migraph/gpu/miopen.hpp>
#include <migraph/gpu/hip.hpp>
#include <migraph/dfor.hpp>
...
...
@@ -18,6 +19,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
hip_mul
...
...
@@ -29,7 +31,7 @@ struct hip_mul
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/pooling.hpp
View file @
4c031df7
...
...
@@ -7,6 +7,7 @@
#include <migraph/operators.hpp>
#include <migraph/generate.hpp>
#include <migraph/shape_for_each.hpp>
#include <migraph/config.hpp>
#include <migraph/gpu/miopen.hpp>
#include <migraph/gpu/hip.hpp>
#include <migraph/dfor.hpp>
...
...
@@ -18,6 +19,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
miopen_pooling
...
...
@@ -33,7 +35,7 @@ struct miopen_pooling
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/relu.hpp
View file @
4c031df7
...
...
@@ -7,6 +7,7 @@
#include <migraph/operators.hpp>
#include <migraph/generate.hpp>
#include <migraph/shape_for_each.hpp>
#include <migraph/config.hpp>
#include <migraph/gpu/miopen.hpp>
#include <migraph/gpu/hip.hpp>
#include <migraph/dfor.hpp>
...
...
@@ -18,6 +19,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
miopen_relu
...
...
@@ -31,7 +33,7 @@ struct miopen_relu
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/rocblas.hpp
View file @
4c031df7
...
...
@@ -3,9 +3,11 @@
#include <migraph/manage_ptr.hpp>
#include <migraph/operators.hpp>
#include <migraph/config.hpp>
#include <rocblas.h>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
using
rocblas_handle_ptr
=
MIGRAPH_MANAGE_PTR
(
rocblas_handle
,
rocblas_destroy_handle
);
...
...
@@ -14,7 +16,7 @@ rocblas_handle_ptr create_rocblas_handle_ptr();
rocblas_handle_ptr
create_rocblas_handle_ptr
(
hipStream_t
s
);
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/softmax.hpp
View file @
4c031df7
...
...
@@ -7,6 +7,7 @@
#include <migraph/operators.hpp>
#include <migraph/generate.hpp>
#include <migraph/shape_for_each.hpp>
#include <migraph/config.hpp>
#include <migraph/gpu/miopen.hpp>
#include <migraph/gpu/hip.hpp>
#include <migraph/dfor.hpp>
...
...
@@ -18,6 +19,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
miopen_softmax
...
...
@@ -31,7 +33,7 @@ struct miopen_softmax
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/target.hpp
View file @
4c031df7
...
...
@@ -2,8 +2,10 @@
#define MIGRAPH_GUARD_MIGRAPHLIB_MIOPEN_TARGET_HPP
#include <migraph/program.hpp>
#include <migraph/config.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
target
...
...
@@ -12,7 +14,9 @@ struct target
std
::
vector
<
pass
>
get_passes
(
migraph
::
context
&
gctx
)
const
;
migraph
::
context
get_context
()
const
;
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/include/migraph/gpu/write_literals.hpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <migraph/gpu/context.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
...
...
@@ -17,7 +18,7 @@ struct write_literals
};
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
#endif
src/targets/gpu/leaky_relu.cpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
shape
miopen_leaky_relu
::
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
...
...
@@ -33,5 +34,5 @@ argument miopen_leaky_relu::compute(context& ctx,
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/lowering.cpp
View file @
4c031df7
...
...
@@ -27,6 +27,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
struct
miopen_apply
...
...
@@ -220,4 +221,5 @@ struct miopen_apply
void
lowering
::
apply
(
program
&
p
)
const
{
miopen_apply
{
&
p
,
ctx
}.
apply
();
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/mul.cpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
shape
hip_mul
::
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
...
...
@@ -21,5 +22,5 @@ argument hip_mul::compute(context& ctx, const shape&, const std::vector<argument
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/pooling.cpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
shape
miopen_pooling
::
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
...
...
@@ -37,5 +38,5 @@ argument miopen_pooling::compute(context& ctx,
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/relu.cpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
shape
miopen_relu
::
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
...
...
@@ -33,5 +34,5 @@ argument miopen_relu::compute(context& ctx,
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/rocblas.cpp
View file @
4c031df7
#include <migraph/gpu/rocblas.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
rocblas_handle_ptr
create_rocblas_handle_ptr
()
...
...
@@ -18,5 +19,5 @@ rocblas_handle_ptr create_rocblas_handle_ptr(hipStream_t s)
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/softmax.cpp
View file @
4c031df7
...
...
@@ -5,6 +5,7 @@
#include <utility>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
shape
miopen_softmax
::
compute_shape
(
const
std
::
vector
<
shape
>&
inputs
)
const
...
...
@@ -32,5 +33,5 @@ argument miopen_softmax::compute(context& ctx,
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/target.cpp
View file @
4c031df7
...
...
@@ -19,6 +19,7 @@
#include <migraph/gpu/concat_gpu_opt.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
std
::
vector
<
pass
>
target
::
get_passes
(
migraph
::
context
&
gctx
)
const
...
...
@@ -60,4 +61,5 @@ std::string target::name() const { return "miopen"; }
migraph
::
context
target
::
get_context
()
const
{
return
context
{};
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
src/targets/gpu/write_literals.cpp
View file @
4c031df7
...
...
@@ -5,7 +5,7 @@
#include <migraph/env.hpp>
namespace
migraph
{
inline
namespace
MIGRAPH_INLINE_NS
{
namespace
gpu
{
MIGRAPH_DECLARE_ENV_VAR
(
MIGRAPH_COPY_LITERALS
)
...
...
@@ -51,5 +51,7 @@ void write_literals::apply(program& p) const
}
}
}
}
// namespace gpu
}
// namespace MIGRAPH_INLINE_NS
}
// namespace migraph
test/auto_contiguous_test.cpp
View file @
4c031df7
...
...
@@ -14,6 +14,7 @@ struct contiguous_target
migraph
::
context
get_context
()
const
{
return
{};
}
};
// TODO: Add this test case
void
literal_broadcast
()
{
migraph
::
program
p
;
...
...
@@ -25,7 +26,7 @@ void literal_broadcast()
EXPECT
(
not
p
.
get_shape
().
broadcasted
());
}
void
literal_transpose
(
)
TEST_CASE
(
literal_transpose
)
{
migraph
::
program
p
;
p
.
add_literal
(
get_2x2_transposed
());
...
...
@@ -36,7 +37,7 @@ void literal_transpose()
EXPECT
(
not
p
.
get_shape
().
transposed
());
}
void
after_literal_transpose
(
)
TEST_CASE
(
after_literal_transpose
)
{
migraph
::
program
p
;
auto
l
=
p
.
add_literal
(
get_2x2
());
...
...
@@ -51,7 +52,7 @@ void after_literal_transpose()
EXPECT
(
not
p
.
get_shape
().
transposed
());
}
void
after_literal_broadcast
(
)
TEST_CASE
(
after_literal_broadcast
)
{
migraph
::
program
p
;
auto
l1
=
p
.
add_literal
(
get_2x2
());
...
...
@@ -67,7 +68,7 @@ void after_literal_broadcast()
EXPECT
(
not
p
.
get_shape
().
broadcasted
());
}
void
after_param_transpose
(
)
TEST_CASE
(
after_param_transpose
)
{
migraph
::
program
p
;
auto
l
=
p
.
add_parameter
(
"2x2"
,
{
migraph
::
shape
::
float_type
,
{
2
,
2
}});
...
...
@@ -82,7 +83,7 @@ void after_param_transpose()
EXPECT
(
not
p
.
get_shape
().
transposed
());
}
void
after_param_broadcast
(
)
TEST_CASE
(
after_param_broadcast
)
{
migraph
::
program
p
;
auto
l1
=
p
.
add_parameter
(
"2x2"
,
{
migraph
::
shape
::
float_type
,
{
2
,
2
}});
...
...
@@ -98,12 +99,4 @@ void after_param_broadcast()
EXPECT
(
not
p
.
get_shape
().
broadcasted
());
}
int
main
()
{
// literal_broadcast();
literal_transpose
();
after_literal_transpose
();
after_literal_broadcast
();
after_param_transpose
();
after_param_broadcast
();
}
int
main
(
int
argc
,
const
char
*
argv
[])
{
test
::
run
(
argc
,
argv
);
}
test/common_subexpression_elimination_test.cpp
View file @
4c031df7
...
...
@@ -14,7 +14,7 @@ struct cse_target
migraph
::
context
get_context
()
const
{
return
{};
}
};
void
cse_test1
(
)
TEST_CASE
(
cse_test1
)
{
migraph
::
program
p1
;
{
...
...
@@ -38,7 +38,7 @@ void cse_test1()
EXPECT
(
p1
==
p2
);
}
void
cse_test2
(
)
TEST_CASE
(
cse_test2
)
{
migraph
::
program
p1
;
{
...
...
@@ -63,7 +63,7 @@ void cse_test2()
EXPECT
(
p1
==
p2
);
}
void
cse_test3
(
)
TEST_CASE
(
cse_test3
)
{
migraph
::
program
p1
;
{
...
...
@@ -86,7 +86,7 @@ void cse_test3()
EXPECT
(
p1
==
p2
);
}
void
cse_test4
(
)
TEST_CASE
(
cse_test4
)
{
migraph
::
program
p1
;
{
...
...
@@ -112,10 +112,4 @@ void cse_test4()
EXPECT
(
p1
==
p2
);
}
int
main
()
{
cse_test1
();
cse_test2
();
cse_test3
();
cse_test4
();
}
int
main
(
int
argc
,
const
char
*
argv
[])
{
test
::
run
(
argc
,
argv
);
}
Prev
1
…
3
4
5
6
7
8
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