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
1d4a7c11
Unverified
Commit
1d4a7c11
authored
Nov 05, 2021
by
Shucai Xiao
Committed by
GitHub
Nov 05, 2021
Browse files
Merge branch 'develop' into test_runner_match_input_output
parents
df2da5de
04e17804
Changes
12
Hide whitespace changes
Inline
Side-by-side
Showing
12 changed files
with
33 additions
and
24 deletions
+33
-24
Dockerfile
Dockerfile
+3
-1
Jenkinsfile
Jenkinsfile
+7
-1
hip-clang.docker
hip-clang.docker
+3
-1
src/api/include/migraphx/migraphx.hpp
src/api/include/migraphx/migraphx.hpp
+11
-12
src/include/migraphx/op/if_op.hpp
src/include/migraphx/op/if_op.hpp
+1
-1
src/include/migraphx/op/loop.hpp
src/include/migraphx/op/loop.hpp
+1
-1
src/include/migraphx/op/topk.hpp
src/include/migraphx/op/topk.hpp
+2
-2
src/include/migraphx/run_loop.hpp
src/include/migraphx/run_loop.hpp
+1
-1
src/include/migraphx/stringutils.hpp
src/include/migraphx/stringutils.hpp
+1
-1
src/targets/cpu/dnnl.cpp
src/targets/cpu/dnnl.cpp
+1
-1
test/msgpack.cpp
test/msgpack.cpp
+1
-1
test/run_loop_test.cpp
test/run_loop_test.cpp
+1
-1
No files found.
Dockerfile
View file @
1d4a7c11
...
...
@@ -6,7 +6,7 @@ ARG PREFIX=/usr/local
RUN
dpkg
--add-architecture
i386
# Add rocm repository
RUN
sh
-c
'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/4.
2
/
xenial
main > /etc/apt/sources.list.d/rocm.list'
RUN
sh
-c
'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/4.
5
/
ubuntu
main > /etc/apt/sources.list.d/rocm.list'
# Install dependencies
RUN
apt-get update
&&
DEBIAN_FRONTEND
=
noninteractive apt-get
install
-y
--allow-unauthenticated
\
...
...
@@ -32,6 +32,8 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --allow-
software-properties-common
\
wget
\
rocm-device-libs
\
hip-base
\
libnuma-dev
\
miopen-hip
\
rocblas
\
zlib1g-dev
&&
\
...
...
Jenkinsfile
View file @
1d4a7c11
...
...
@@ -20,7 +20,7 @@ def rocmtestnode(Map conf) {
rm -rf build
mkdir build
cd build
CXX=${compiler} CXXFLAGS='-Werror
-Wno-fallback
' cmake -DCMAKE_C_COMPILER_LAUNCHER=ccache -DCMAKE_CXX_COMPILER_LAUNCHER=ccache ${flags} ..
CXX=${compiler} CXXFLAGS='-Werror' cmake -DCMAKE_C_COMPILER_LAUNCHER=ccache -DCMAKE_CXX_COMPILER_LAUNCHER=ccache ${flags} ..
make -j\$(nproc) generate all doc package check VERBOSE=1
"""
echo
cmd
...
...
@@ -75,6 +75,8 @@ def rocmnodename(name) {
node_name
=
"${rocmtest_name} && fiji"
;
}
else
if
(
name
==
"vega"
)
{
node_name
=
"${rocmtest_name} && vega"
;
}
else
if
(
name
==
"navi21"
)
{
node_name
=
"${rocmtest_name} && navi21"
;
}
else
if
(
name
==
"nogpu"
)
{
return
rocmtest_name
;
}
...
...
@@ -110,6 +112,10 @@ rocmtest clang_debug: rocmnode('vega') { cmake_build ->
def
debug_flags
=
"-g -O2 -fno-omit-frame-pointer -fsanitize=${sanitizers} -fno-sanitize-recover=${sanitizers}"
cmake_build
(
"/opt/rocm/llvm/bin/clang++"
,
"-DCMAKE_BUILD_TYPE=debug -DMIGRAPHX_ENABLE_PYTHON=Off -DMIGRAPHX_ENABLE_GPU=Off -DMIGRAPHX_ENABLE_CPU=On -DCMAKE_CXX_FLAGS_DEBUG='${debug_flags}'"
)
}
},
clang_release_navi:
rocmnode
(
'navi21'
)
{
cmake_build
->
stage
(
'HIP Clang Release Navi'
)
{
cmake_build
(
"/opt/rocm/llvm/bin/clang++"
,
"-DCMAKE_BUILD_TYPE=release"
)
}
}
def
onnxnode
(
name
,
body
)
{
...
...
hip-clang.docker
View file @
1d4a7c11
...
...
@@ -6,7 +6,7 @@ ARG PREFIX=/usr/local
RUN
dpkg
--add-architecture
i386
# Add rocm repository
RUN
sh
-c
'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/4.
2
/
xenial
main > /etc/apt/sources.list.d/rocm.list'
RUN
sh
-c
'echo deb [arch=amd64 trusted=yes] http://repo.radeon.com/rocm/apt/4.
5
/
ubuntu
main > /etc/apt/sources.list.d/rocm.list'
# Install dependencies
RUN
apt-get update
&&
DEBIAN_FRONTEND
=
noninteractive apt-get
install
-y
--allow-unauthenticated
\
...
...
@@ -29,6 +29,8 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --allow-
software-properties-common
\
wget
\
rocm-device-libs
\
hip-base
\
libnuma-dev
\
miopen-hip
\
rocblas
\
zlib1g-dev
&&
\
...
...
src/api/include/migraphx/migraphx.hpp
View file @
1d4a7c11
...
...
@@ -252,7 +252,7 @@ struct shape : MIGRAPHX_CONST_HANDLE_BASE(shape)
const
size_t
*
pout
;
size_t
pout_size
;
call
(
&
migraphx_shape_lengths
,
&
pout
,
&
pout_size
,
this
->
get_handle_ptr
());
return
std
::
vector
<
size_t
>
(
pout
,
pout
+
pout_size
)
;
return
{
pout
,
pout
+
pout_size
}
;
}
std
::
vector
<
size_t
>
strides
()
const
...
...
@@ -260,7 +260,7 @@ struct shape : MIGRAPHX_CONST_HANDLE_BASE(shape)
const
size_t
*
pout
;
size_t
pout_size
;
call
(
&
migraphx_shape_strides
,
&
pout
,
&
pout_size
,
this
->
get_handle_ptr
());
return
std
::
vector
<
size_t
>
(
pout
,
pout
+
pout_size
)
;
return
{
pout
,
pout
+
pout_size
}
;
}
migraphx_shape_datatype_t
type
()
const
...
...
@@ -312,7 +312,7 @@ struct argument : MIGRAPHX_CONST_HANDLE_BASE(argument)
{
const_migraphx_shape_t
pout
;
call
(
&
migraphx_argument_shape
,
&
pout
,
this
->
get_handle_ptr
());
return
shape
(
pout
)
;
return
{
pout
}
;
}
char
*
data
()
const
...
...
@@ -325,9 +325,8 @@ struct argument : MIGRAPHX_CONST_HANDLE_BASE(argument)
/// Generate an argument using random data
static
argument
generate
(
shape
ps
,
size_t
pseed
=
0
)
{
return
argument
(
make
<
migraphx_argument
>
(
&
migraphx_argument_generate
,
ps
.
get_handle_ptr
(),
pseed
),
own
{});
return
{
make
<
migraphx_argument
>
(
&
migraphx_argument_generate
,
ps
.
get_handle_ptr
(),
pseed
),
own
{}};
}
friend
bool
operator
==
(
const
argument
&
px
,
const
argument
&
py
)
...
...
@@ -378,7 +377,7 @@ struct program_parameter_shapes : MIGRAPHX_HANDLE_BASE(program_parameter_shapes)
{
const_migraphx_shape_t
pout
;
call
(
&
migraphx_program_parameter_shapes_get
,
&
pout
,
this
->
get_handle_ptr
(),
pname
);
return
shape
(
pout
)
;
return
{
pout
}
;
}
std
::
vector
<
const
char
*>
names
()
const
...
...
@@ -438,7 +437,7 @@ struct arguments : MIGRAPHX_HANDLE_BASE(arguments), array_base<arguments>
{
const_migraphx_argument_t
pout
;
call
(
&
migraphx_arguments_get
,
&
pout
,
this
->
get_handle_ptr
(),
pidx
);
return
argument
(
pout
)
;
return
{
pout
}
;
}
struct
iterator_read
...
...
@@ -449,7 +448,7 @@ struct arguments : MIGRAPHX_HANDLE_BASE(arguments), array_base<arguments>
const_migraphx_argument_t
pout
;
call
(
&
migraphx_arguments_get
,
&
pout
,
self
,
pidx
);
return
argument
(
pout
)
;
return
{
pout
}
;
}
};
};
...
...
@@ -471,7 +470,7 @@ struct shapes : MIGRAPHX_HANDLE_BASE(shapes), array_base<shapes>
{
const_migraphx_shape_t
pout
;
call
(
&
migraphx_shapes_get
,
&
pout
,
this
->
get_handle_ptr
(),
pidx
);
return
shape
(
pout
)
;
return
{
pout
}
;
}
struct
iterator_read
...
...
@@ -481,7 +480,7 @@ struct shapes : MIGRAPHX_HANDLE_BASE(shapes), array_base<shapes>
{
const_migraphx_shape_t
pout
;
call
(
&
migraphx_shapes_get
,
&
pout
,
self
,
pidx
);
return
shape
(
pout
)
;
return
{
pout
}
;
}
};
};
...
...
@@ -609,7 +608,7 @@ struct operation : MIGRAPHX_HANDLE_BASE(operation)
{
std
::
array
<
char
,
1024
>
out_name
;
call
(
&
migraphx_operation_name
,
out_name
.
data
(),
1024
,
this
->
get_handle_ptr
());
return
std
::
string
(
out_name
.
data
()
)
;
return
{
out_name
.
data
()
}
;
}
};
...
...
src/include/migraphx/op/if_op.hpp
View file @
1d4a7c11
...
...
@@ -35,7 +35,7 @@ struct if_op
MIGRAPHX_THROW
(
"IF: output shapes of submodules must be the same."
);
}
return
shape
(
out_shapes0
)
;
return
{
out_shapes0
}
;
}
argument
compute
(
const
shape
&
,
...
...
src/include/migraphx/op/loop.hpp
View file @
1d4a7c11
...
...
@@ -54,7 +54,7 @@ struct loop
ins_out_shapes
.
push_back
({
out_s
.
type
(),
lens
});
}
return
shape
(
ins_out_shapes
)
;
return
{
ins_out_shapes
}
;
}
struct
ref_loop
...
...
src/include/migraphx/op/topk.hpp
View file @
1d4a7c11
...
...
@@ -45,7 +45,7 @@ struct topk
shape
s_val
{
type
,
lens
};
shape
s_ind
{
shape
::
int64_type
,
lens
};
return
shape
(
{
s_val
,
s_ind
}
)
;
return
{
{
s_val
,
s_ind
}
}
;
}
template
<
class
T
,
class
Compare
>
...
...
@@ -131,7 +131,7 @@ struct topk
});
});
return
argument
(
{
res_val
,
res_ind
}
)
;
return
{
{
res_val
,
res_ind
}
}
;
}
};
...
...
src/include/migraphx/run_loop.hpp
View file @
1d4a7c11
...
...
@@ -106,7 +106,7 @@ argument run_loop(const LoopModel& model,
std
::
copy
(
in_args
.
begin
()
+
2
,
in_args
.
end
(),
out_args
.
begin
());
model
.
set_zero
(
ctx
,
scan_outputs
,
iter
);
return
argument
(
out_args
)
;
return
{
out_args
}
;
}
}
// namespace MIGRAPHX_INLINE_NS
...
...
src/include/migraphx/stringutils.hpp
View file @
1d4a7c11
...
...
@@ -71,7 +71,7 @@ std::string trim(const std::string& s, F f)
{
auto
start
=
std
::
find_if_not
(
s
.
begin
(),
s
.
end
(),
f
);
auto
last
=
std
::
find_if_not
(
s
.
rbegin
(),
std
::
string
::
const_reverse_iterator
(
start
),
f
).
base
();
return
std
::
string
(
start
,
last
)
;
return
{
start
,
last
}
;
}
inline
std
::
string
trim
(
const
std
::
string
&
s
)
...
...
src/targets/cpu/dnnl.cpp
View file @
1d4a7c11
...
...
@@ -73,7 +73,7 @@ dnnl::memory::desc to_dnnl_memory_desc(const shape& s)
dnnl
::
memory
to_dnnl_memory
(
const
dnnl
::
memory
::
desc
&
desc
,
const
argument
&
a
)
{
return
dnnl
::
memory
(
desc
,
get_dnnl_context
().
engine
,
a
.
data
()
)
;
return
{
desc
,
get_dnnl_context
().
engine
,
a
.
data
()
}
;
}
dnnl
::
memory
to_dnnl_memory
(
const
argument
&
a
)
...
...
test/msgpack.cpp
View file @
1d4a7c11
...
...
@@ -11,7 +11,7 @@ std::vector<char> msgpack_buffer(const T& src)
msgpack
::
pack
(
buffer
,
src
);
buffer
.
seekg
(
0
);
std
::
string
str
=
buffer
.
str
();
return
std
::
vector
<
char
>
(
str
.
data
(),
str
.
data
()
+
str
.
size
());
return
std
::
vector
<
char
>
(
str
.
data
(),
str
.
data
()
+
str
.
size
());
// NOLINT
}
TEST_CASE
(
test_msgpack_empty_value
)
...
...
test/run_loop_test.cpp
View file @
1d4a7c11
...
...
@@ -75,7 +75,7 @@ struct test_loop_op
ins_out_shapes
.
push_back
({
out_s
.
type
(),
lens
});
}
return
migraphx
::
shape
(
ins_out_shapes
)
;
return
{
ins_out_shapes
}
;
}
struct
test_loop
:
public
migraphx
::
op
::
loop
::
ref_loop
...
...
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